blob: 35cb7993e1d3df975f09d3b1650c3faac4223849 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
|
/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
|