File tree 1 file changed +7
-4
lines changed 1 file changed +7
-4
lines changed Original file line number Diff line number Diff line change @@ -855,11 +855,14 @@ JIT::JIT(const std::string &libidtr)
855
855
if (!gpuxlibstr.empty ()) {
856
856
_gpulib = std::string (gpuxlibstr);
857
857
} else {
858
- auto imexRoot = get_text_env (" IMEXROOT" );
859
- imexRoot = !imexRoot.empty () ? imexRoot : std::string (CMAKE_IMEX_ROOT);
860
- _gpulib = imexRoot + " /lib/liblevel-zero-runtime.so" ;
858
+ // auto imexRoot = get_text_env("IMEXROOT");
859
+ // imexRoot = !imexRoot.empty() ? imexRoot : std::string(CMAKE_IMEX_ROOT);
860
+ // _gpulib = imexRoot + "/lib/liblevel-zero-runtime.so";
861
+ // _gpulib = imexRoot + "/lib/liblevel-zero-runtime.so";
862
+ // for nv gpu
863
+ _gpulib = mlirRoot + " /lib/libmlir_cuda_runtime.so" ;
861
864
if (!std::ifstream (_gpulib)) {
862
- throw std::runtime_error (" Cannot find liblevel-zero-runtime.so " );
865
+ throw std::runtime_error (" Cannot find lib: " + _gpulib );
863
866
}
864
867
}
865
868
_sharedLibPaths = {_crunnerlib.c_str (), _runnerlib.c_str (),
You can’t perform that action at this time.
0 commit comments