blob: a057dd466bcaf07f74cb14ede86a96dc2eedc945 (
plain)
1
2
3
4
5
6
7
|
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
|