Fix gpu list
Signed-off-by: Tom Rix <Tom.Rix@amd.com>
This commit is contained in:
parent
33eff2db68
commit
470a4ad4df
1 changed files with 2 additions and 2 deletions
|
|
@ -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
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue