Skip to content

Use MAGMA option for PyTorch 2+ #3683

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
Flamefire wants to merge 3 commits into
base: develop
Choose a base branch
from
+10 −4
Open

Use MAGMA option for PyTorch 2+ #3683

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
6bf10a7
Use MAGMA option for PyTorch 2+
Flamefire Apr 2, 2025
53f9e62
Remove version prefix from comparison
Flamefire May 28, 2025
ca9cdcc
Merge branch 'develop' into 20250402122705_new_pr_pytorch
Flamefire Feb 24, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
14 changes: 10 additions & 4 deletions easybuild/easyblocks/p/pytorch.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -509,14 +509,20 @@ def add_enable_option(name, enabled):
# determine the compute capability of a GPU in the system and use that which may fail tests if
# it is to new for the used nvcc
env.setvar('TORCH_CUDA_ARCH_LIST', ';'.join(cuda_cc))
with_gpu_support = True
else:
# Disable CUDA
options.append('USE_CUDA=0')
with_gpu_support = False

if pytorch_version >= '2.0':
add_enable_option('USE_ROCM', get_software_root('ROCm'))
elif pytorch_version >= 'v1.10.0':
add_enable_option('USE_MAGMA', get_software_root('magma'))
has_rocm = get_software_root('ROCm')
add_enable_option('USE_ROCM', has_rocm)
if has_rocm:
with_gpu_support = True

if pytorch_version >= '1.10.0':
add_enable_option('USE_MAGMA', with_gpu_support and get_software_root('magma'))

if get_cpu_architecture() == POWER:
# *NNPACK is not supported on Power, disable to avoid warnings
Expand All @@ -525,7 +531,7 @@ def add_enable_option(name, enabled):
if pytorch_version >= '1.10.0' and pytorch_version < '1.12.0':
options.append('USE_BREAKPAD=0')
# FBGEMM requires AVX512, so not available on PPC
if pytorch_version >= 'v1.10.0':
if pytorch_version >= '1.10.0':
options.append('USE_FBGEMM=0')

# Metal only supported on IOS which likely doesn't work with EB, so disabled
Expand Down