Merge branch 'master' into reduce_global_var_use_in_checkpoiting_to_r…

…educe_compile_graph_breaks

op_builder/builder.py

@@ -253,8 +253,7 @@ def get_rocm_gpu_arch():
  rocm_info = Path("rocminfo")
  rocm_gpu_arch_cmd = str(rocm_info) + " | grep -o -m 1 'gfx.*'"
  try:
- safe_cmd = shlex.split(rocm_gpu_arch_cmd)
- result = subprocess.check_output(safe_cmd)
+ result = subprocess.check_output(rocm_gpu_arch_cmd, shell=True)
  rocm_gpu_arch = result.decode('utf-8').strip()
  except subprocess.CalledProcessError:
  rocm_gpu_arch = ""
@@ -272,8 +271,7 @@ def get_rocm_wavefront_size():
  rocm_wavefront_size_cmd = str(
  rocm_info) + " | grep -Eo -m1 'Wavefront Size:[[:space:]]+[0-9]+' | grep -Eo '[0-9]+'"
  try:
- safe_cmd = shlex.split(rocm_wavefront_size_cmd)
- result = subprocess.check_output(rocm_wavefront_size_cmd)
+ result = subprocess.check_output(rocm_wavefront_size_cmd, shell=True)
  rocm_wavefront_size = result.decode('utf-8').strip()
  except subprocess.CalledProcessError:
  rocm_wavefront_size = "32"