From 2c0e2ef19fa8e62081cadb267e1cf1ee77e501a7 Mon Sep 17 00:00:00 2001 From: Masahiro Masuda Date: Sat, 22 Dec 2018 20:24:41 +0900 Subject: [PATCH] [ROCM] Make sure all bit code files exist --- python/tvm/contrib/rocm.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/python/tvm/contrib/rocm.py b/python/tvm/contrib/rocm.py index 10cfaed83e68..172d081ff96a 100644 --- a/python/tvm/contrib/rocm.py +++ b/python/tvm/contrib/rocm.py @@ -1,6 +1,6 @@ """Utility for ROCm backend""" import subprocess -from os.path import join +from os.path import join, exists from . import util from .._ffi.base import py_str from ..api import register_func, convert @@ -79,4 +79,5 @@ def callback_rocm_bitcode_path(rocdl_dir="/opt/rocm/lib/"): "oclc_unsafe_math_off.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)])