We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent abb107e commit 999ae8bCopy full SHA for 999ae8b
.github/workflows/build_wheels_rocm.yml
@@ -89,6 +89,10 @@ jobs:
89
$env:PYTORCH_ROCM_ARCH = 'gfx803;gfx900;gfx906:xnack-;gfx908:xnack-;gfx90a:xnack+;gfx90a:xnack-;gfx1010;gfx1012;gfx1030;gfx1100;gfx1101;gfx1102'
90
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'}
91
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
+
96
python setup.py sdist bdist_wheel
97
98
- uses: actions/upload-artifact@v3
0 commit comments