batch  cos_orig  cos_fast       res
/home/yurenh2/miniconda3/lib/python3.13/site-packages/torch/autograd/graph.py:865: UserWarning: Attempting to run cuBLAS, but there was no current CUDA context! Attempting to set the primary context... (Triggered internally at /pytorch/aten/src/ATen/cuda/CublasHandlePool.cpp:330.)
  return Variable._execution_engine.run_backward(  # Calls into the C++ engine to run the backward pass
    0    0.9068    0.9117  9.47e-03
    1    0.8533    0.8525  4.00e-03
    2    0.9177    0.9178  4.00e-03
HOLO_FAST_GATE_DONE