From 1b2e553b1baec308fcdead5462ee1c8248fcff69 Mon Sep 17 00:00:00 2001
From: masahi <masahi129@gmail.com>
Date: Tue, 25 Dec 2018 04:05:29 +0900
Subject: [PATCH] [ROCM] Make sure all bit code files exist (#2323)
---
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 10cfaed83..172d081ff 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)])
--
GitLab