Formatting fix

microsoft · May 14, 2024 · c87c4f3 · c87c4f3
1 parent 88c7be4
commit c87c4f3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/op_builder/builder.py b/op_builder/builder.py
@@ -248,7 +248,7 @@ def get_rocm_gpu_arch():
     def get_rocm_wavefront_size():
         if OpBuilder._rocm_wavefront_size:
             return OpBuilder._rocm_wavefront_size
-        rocm_wavefront_size_cmd = "/opt/rocm/bin/rocminfo | grep -Eo -m1 'Wavefront Size:\s+[0-9]+' | grep -Eo '[0-9]+'"
+        rocm_wavefront_size_cmd = "/opt/rocm/bin/rocminfo | grep -Eo -m1 'Wavefront Size:[[:space:]]+[0-9]+' | grep -Eo '[0-9]+'"
         try:
             result = subprocess.check_output(rocm_wavefront_size_cmd, shell=True)
             rocm_wavefront_size = result.decode('utf-8').strip()