From afaee8601a89fc12da7dbc97aeefe3aca1687bd4 Mon Sep 17 00:00:00 2001 From: "rraminen@amd.com" Date: Thu, 18 Apr 2024 21:10:18 +0000 Subject: [PATCH] Formatting fix --- op_builder/builder.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/op_builder/builder.py b/op_builder/builder.py index 9042f8c7caccc..18c130221b0e7 100644 --- 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()