Commit 1b2e553b by masahi Committed by Tianqi Chen

[ROCM] Make sure all bit code files exist (#2323)

parent d5164466
"""Utility for ROCm backend""" """Utility for ROCm backend"""
import subprocess import subprocess
from os.path import join from os.path import join, exists
from . import util from . import util
from .._ffi.base import py_str from .._ffi.base import py_str
from ..api import register_func, convert from ..api import register_func, convert
...@@ -79,4 +79,5 @@ def callback_rocm_bitcode_path(rocdl_dir="/opt/rocm/lib/"): ...@@ -79,4 +79,5 @@ def callback_rocm_bitcode_path(rocdl_dir="/opt/rocm/lib/"):
"oclc_unsafe_math_off.amdgcn.bc", "oclc_unsafe_math_off.amdgcn.bc",
"oclc_unsafe_math_on.amdgcn.bc" "oclc_unsafe_math_on.amdgcn.bc"
] ]
return convert([join(rocdl_dir, bitcode) for bitcode in bitcode_files]) paths = [join(rocdl_dir, bitcode) for bitcode in bitcode_files]
return convert([path for path in paths if exists(path)])
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment