work rocm

wuxxin · Oct 22, 2022 · c0d31f0 · c0d31f0
1 parent fc1cf26
commit c0d31f0
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 16 deletions.
diff --git a/hardware/amd/rocm/init.sls b/hardware/amd/rocm/init.sls
@@ -2,14 +2,15 @@
 
 include:
   - manjaro.aur
+  # - manjaro.arch4edu
   - hardware.amd.radeon
 
 {{ pamac_install('rocm-sdk', [
   'rocminfo',
   'rocm-hip-sdk',
   'rocm-opencl-sdk',
   'miopen-hip',
-  'hipmagma',
+  'hip-runtime-amd',
   ]) }}
 
 hardware-amd-rocm:

diff --git a/hardware/amd/rocm/jax.sls b/hardware/amd/rocm/jax.sls
@@ -12,5 +12,5 @@ jax_req:
       - sls: desktop.manjaro.python.development
       - sls: desktop.manjaro.python.hardware_optimized
 
-{{ pamac_install('python-jax-rocm_aur', ['python-jax-rocm'],
+{{ pamac_install('python-jax-rocm-aur', ['python-jax-rocm'],
     require= "test: jax_req") }}
diff --git a/hardware/amd/rocm/pytorch.sls b/hardware/amd/rocm/pytorch.sls
@@ -13,7 +13,7 @@ pytorch_req:
       - sls: desktop.manjaro.python.hardware_optimized
 
 # pytorch - Tensors and Dynamic neural networks in Python with strong GPU acceleration
-{{ pamac_install('python-pytorch-rocm', ['python-pytorch-rocm', ],
+{{ pamac_install('python-pytorch-rocm', ['python-pytorch-opt-rocm', ],
     require='test: pytorch_req') }}
 
 # torchvision - Datasets, transforms, and models specific to computer vision
@@ -25,18 +25,7 @@ pytorch_req:
     require='test: pytorch_req') }}
 
 {#
-
-# torchtext - Data loaders and abstractions for text and NLP
-{{ pamac_patch_install_dir('python-torchtext',
-    'salt://hardware/amd/rocm/python-torchtext',
-    require='test: python-pytorch-rocm', custom=true) }}
-
 # torchaudio - Data manipulation and transformation for audio signal processing
-{{ pamac_patch_install_dir('python-torchaudio-rocm',
-    'salt://hardware/amd/rocm/python-torchaudio-rocm',
-    require=['test: python-pytorch-rocm', 'test: python-kaldi-io'],
-    custom=true) }}
-
+# torchtext - Data loaders and abstractions for text and NLP
 # torchrec - Domain library for recommendation systems
-
 #}
diff --git a/hardware/amd/rocm/tensorflow.sls b/hardware/amd/rocm/tensorflow.sls
@@ -13,5 +13,5 @@ tensorflow_req:
       - sls: desktop.manjaro.python.hardware_optimized
 
 # python-tensorflow-opt-rocm - scalable machine learning (with ROCM and AVX2 CPU optimizations)
-{{ pamac_install('python-tensorflow-rocm_aur', [ 'python-tensorflow-opt-rocm',],
+{{ pamac_install('python-tensorflow-rocm-aur', [ 'python-tensorflow-opt-rocm',],
     require= 'test: tensorflow_req') }}