diff --git a/代码/code/build_env.sh b/代码/code/build_env.sh index 5ade4cd..74fff2d 100644 --- a/代码/code/build_env.sh +++ b/代码/code/build_env.sh @@ -1,17 +1,2 @@ #!/bin/bash -set -e - -# 预热 torch inductor,避免推理时编译 -python -c " -import torch - -@torch.compile(mode='reduce-overhead') -def _warmup(x): - return x * 2 - -x = torch.randn(100, 100, device='cuda') -_warmup(x) -print('Inductor cache ready') -" - echo "build env success"