Skip to content

Commit f937a46

Browse files
committed
Fix rocm
1 parent 8adfa85 commit f937a46

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

workspace.bzl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -106,7 +106,7 @@ sed -i.bak0 "s/tf_http_archive(/http_archive(/g" third_party/py/python_init_rule
106106
""",
107107
"""
108108
sed -i.bak0 "s/_DEFAULT_ROCM_TOOLKIT_PATH =/_TMPDIR = \\"TMPDIR\\"\\n_DEFAULT_ROCM_TOOLKIT_PATH =/g" third_party/gpus/rocm_configure.bzl
109-
sed -i.bak0 "s/\\"%{rocm_hipblaslt}\\"/\\"%{tmpdir}\\":get_host_environ(repository_ctx,_TMPDIR,\\"\\"), \\"%{rocm_hipblaslt}\\"/g" third_party/gpus/rocm_configure.bzl
109+
sed -i.bak0 "s/\\"%{gcc_host_compiler_path}\\"/\\"%{tmpdir}\\":get_host_environ(repository_ctx,_TMPDIR,\\"\\"), \\"%{gcc_host_compiler_path}\\"/g" third_party/gpus/rocm_configure.bzl
110110
sed -i.bak0 "s/VERBOSE =/TMPDIR= '%{tmpdir}'\\nVERBOSE =/g" third_party/gpus/crosstool/clang/bin/crosstool_wrapper_driver_rocm.tpl
111111
sed -i.bak0 "s/def main():/def main():\\n if TMPDIR: os.environ['TMPDIR'] = TMPDIR/g" third_party/gpus/crosstool/clang/bin/crosstool_wrapper_driver_is_not_gcc.tpl third_party/gpus/crosstool/clang/bin/crosstool_wrapper_driver_rocm.tpl
112112
""",

0 commit comments

Comments
 (0)