Update gpu_burn.py

Define hip and cuda as environment feature

Author: CedricJour

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')