Skip to content

Commit 5cce2db

Browse files
authored
Another attempt to use mkl_free_buffers (#418)
1 parent f20d471 commit 5cce2db

File tree

1 file changed

+2
-10
lines changed

1 file changed

+2
-10
lines changed

deps/src/onemkl.cpp

Lines changed: 2 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -4319,18 +4319,10 @@ extern "C" int onemklXsparse_matmat(syclQueue_t device_queue, matrix_handle_t A,
43194319
// other
43204320

43214321
// oneMKL keeps a cache of SYCL queues and tries to destroy them when unloading the library.
4322-
// that is incompatible with oneAPI.jl destroying queues before that, so expose a function
4322+
// that is incompatible with oneAPI.jl destroying queues before that, so call mkl_free_buffers
43234323
// to manually wipe the device cache when we're destroying queues.
43244324

4325-
namespace oneapi {
4326-
namespace mkl {
4327-
namespace gpu {
4328-
int clean_gpu_caches();
4329-
}
4330-
}
4331-
}
4332-
43334325
extern "C" int onemklDestroy() {
4334-
oneapi::mkl::gpu::clean_gpu_caches();
4326+
mkl_free_buffers();
43354327
return 0;
43364328
}

0 commit comments

Comments
 (0)