Update gpu_burn.py

Define hip and cuda as environment feature
reframe-hpc · Oct 4, 2024 · 79046da · 79046da
1 parent 7a1e65f
commit 79046da
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/hpctestlib/microbenchmarks/gpu/gpu_burn.py b/hpctestlib/microbenchmarks/gpu/gpu_burn.py
@@ -52,9 +52,9 @@ def setup_build(self):
 
         if self.gpu_build is None:
             # Try to set the build type from the partition features
-            if 'cuda' in curr_part.features:
+            if 'cuda' in curr_env.features:
                 self.gpu_build = 'cuda'
-            elif 'hip' in curr_part.features:
+            elif 'hip' in curr_env.features:
                 self.gpu_build = 'hip'
 
         gpu_devices = curr_part.select_devices('gpu')