Skip to content

Commit f0b0277

Browse files
committed
use partition attribute instead of env attribute
1 parent a952b44 commit f0b0277

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

hpctestlib/microbenchmarks/gpu/gpu_burn.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -52,9 +52,9 @@ def setup_build(self):
5252

5353
if self.gpu_build is None:
5454
# Try to set the build type from the partition features
55-
if 'cuda' in curr_env.features:
55+
if 'cuda' in curr_part.features:
5656
self.gpu_build = 'cuda'
57-
elif 'hip' in curr_env.features:
57+
elif 'hip' in curr_part.features:
5858
self.gpu_build = 'hip'
5959

6060
gpu_devices = curr_part.select_devices('gpu')

0 commit comments

Comments
 (0)