Skip to content

Commit 999ae8b

Browse files
committed
Fix ROCm workflow
1 parent abb107e commit 999ae8b

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

.github/workflows/build_wheels_rocm.yml

+4
Original file line numberDiff line numberDiff line change
@@ -89,6 +89,10 @@ jobs:
8989
$env:PYTORCH_ROCM_ARCH = 'gfx803;gfx900;gfx906:xnack-;gfx908:xnack-;gfx90a:xnack+;gfx90a:xnack-;gfx1010;gfx1012;gfx1030;gfx1100;gfx1101;gfx1102'
9090
if ([version]$env:ROCM_VERSION -lt [version]'5.5') {$env:PYTORCH_ROCM_ARCH = 'gfx803;gfx900;gfx906:xnack-;gfx908:xnack-;gfx90a:xnack+;gfx90a:xnack-;gfx1010;gfx1012;gfx1030'}
9191
92+
$setup = Get-Content setup.py -raw
93+
$macroString = "extra_link_args=extra_link_args,`n define_macros=[('HIPBLAS_USE_HIP_HALF', '1')]"
94+
New-Item setup.py -itemType File -value $setup.replace('extra_link_args=extra_link_args', $macroString) -force
95+
9296
python setup.py sdist bdist_wheel
9397
9498
- uses: actions/upload-artifact@v3

0 commit comments

Comments
 (0)