diff --git a/python-torch.spec b/python-torch.spec index 72e4a26..8fddf36 100644 --- a/python-torch.spec +++ b/python-torch.spec @@ -741,8 +741,8 @@ export ROCM_PATH=`hipconfig -R` # pytorch uses clang, not hipcc export HIP_CLANG_PATH=%{rocmllvm_bindir} -# export PYTORCH_ROCM_ARCH=%{rocm_gpu_list_default} -export PYTORCH_ROCM_ARCH=gfx1151 +export PYTORCH_ROCM_ARCH=%{rocm_gpu_list_default} +# export PYTORCH_ROCM_ARCH=gfx1151 %endif