/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 base: 4.50s/step (res 8.2e-03) hf: 3.49s/step (res 8.2e-03) sd: 4.07s/step (res 8.2e-03) hf+sd: 3.06s/step (res 8.2e-03) hf+sd+t80: 4.97s/step (res 8.2e-03) hf+sd+t80+avg: 4.97s/step (res 8.2e-03) cmp: 4.03s/step (res 8.2e-03) /home/yurenh2/miniconda3/lib/python3.13/site-packages/torch/_inductor/compile_fx.py:321: UserWarning: TensorFloat32 tensor cores for float32 matrix multiplication available but not enabled. Consider setting `torch.set_float32_matmul_precision('high')` for better performance. warnings.warn( FULL(cmp_sdpa): 4.77s/step (res 8.2e-03) AGG_BENCH_DONE