diff --git a/examples/CMakeLists.txt b/examples/CMakeLists.txt index 1d37495b858..e99f3d55c6e 100644 --- a/examples/CMakeLists.txt +++ b/examples/CMakeLists.txt @@ -60,14 +60,6 @@ if(DNNL_CPU_SYCL) endforeach() endif() -# Skip examples for CUDA since USM is a default model for the library which is -# not yet supported for Nvidia backend. -if(DNNL_SYCL_CUDA) - foreach(f ${sources}) - list(REMOVE_ITEM sources "${f}") - endforeach() -endif() - foreach(src ${sources}) file(RELATIVE_PATH src_rel_path ${CMAKE_CURRENT_SOURCE_DIR} ${src}) string(REGEX REPLACE "[/_\\.]" "-" example_name ${src_rel_path}) diff --git a/src/gpu/nvidia/README.md b/src/gpu/nvidia/README.md index 301c2384400..9747634d333 100644 --- a/src/gpu/nvidia/README.md +++ b/src/gpu/nvidia/README.md @@ -27,7 +27,7 @@ cmake -DDNNL_CPU_RUNTIME=DPCPP -DDNNL_GPU_RUNTIME=DPCPP \ ## Memory -Currently, only the buffer-based oneDNN API is supported for Nvidia backend. +Both buffer-based and USM-based oneDNN APIs are supported for Nvidia backend. ## Suported Data Types diff --git a/tests/benchdnn/dnnl_memory.cpp b/tests/benchdnn/dnnl_memory.cpp index 87585e838df..310a6d3f9e8 100644 --- a/tests/benchdnn/dnnl_memory.cpp +++ b/tests/benchdnn/dnnl_memory.cpp @@ -113,15 +113,6 @@ int dnn_mem_t::reorder(const dnn_mem_t &rhs, const_dnnl_primitive_attr_t attr) { int dnn_mem_t::initialize_memory_create_sycl(const handle_info_t &handle_info) { #ifdef DNNL_WITH_SYCL - if (is_nvidia_gpu(engine_)) { - // USM is not supported with Nvidia so ignore memory_kind and - // force SYCL buffers. - DNN_SAFE(dnnl_sycl_interop_memory_create(&m_, &md_, engine_, - dnnl_sycl_interop_buffer, handle_info.ptr), - CRIT); - return OK; - } - if (handle_info.is_host_ptr) { // Ignore memory_kind with host pointers and force USM. DNN_SAFE(dnnl_sycl_interop_memory_create(&m_, &md_, engine_, diff --git a/tests/gtests/CMakeLists.txt b/tests/gtests/CMakeLists.txt index e05bb4c27c7..8fbf9a71f75 100644 --- a/tests/gtests/CMakeLists.txt +++ b/tests/gtests/CMakeLists.txt @@ -96,7 +96,6 @@ file(GLOB PRIM_TEST_CASES_SRC test_matmul.cpp test_resampling.cpp test_reduction.cpp - test_softmax_v2.cpp ) if(DNNL_CPU_RUNTIME STREQUAL "NONE") @@ -237,7 +236,7 @@ endif() foreach(TEST_FILE ${PRIM_TEST_CASES_SRC}) get_filename_component(exe ${TEST_FILE} NAME_WE) - if(NOT ${exe} MATCHES "${skip_usm_pattern}" AND NOT DNNL_SYCL_CUDA) + if(NOT ${exe} MATCHES "${skip_usm_pattern}") register_gtest(${exe} ${TEST_FILE}) endif()