[GPU] Disable unused oneDNN features to reduce binary size (#12987)

* [GPU] Disable unused oneDNN features to reduce binary size

* [GPU] Fixed list args

* [GPU] unit test skip

* [GPU] Disable more onednn primitives
This commit is contained in:
Vladimir Paramuzov authored and GitHub committed 2022-11-29 18:55:14 +04:00
1 parent c4e15e0883
commit a1e30996a4
2 files changed
+11

No files matched your search

@@ -1717,6 +1717,8 @@ TEST(fully_connected_3d_onednn_gpu, no_biases_int8) {
output_b = 2, output_f = 4;
auto& engine = get_onednn_test_engine();
if (!engine.get_device_info().supports_immad)
return;
auto input_prim = engine.allocate_memory({ data_types::f32, format::bfyx, { input_b, input_f, 1, input_y } });
auto weights_prim = engine.allocate_memory({ data_types::i8, format::bfyx, { weight_o, weight_i, 1, 1 } });
+9
View File
@@ -20,6 +20,8 @@ if(ENABLE_ONEDNN_FOR_GPU)
set(ONEDNN_BUILD_DIR "${CMAKE_CURRENT_BINARY_DIR}/onednn_gpu_build/")
set(ONEDNN_INSTALL_DIR "${CMAKE_CURRENT_BINARY_DIR}/onednn_gpu_install/")
set(ONEDNN_PREFIX_DIR "${CMAKE_CURRENT_BINARY_DIR}/onednn_gpu_root")
set(ONEDNN_ENABLED_PRIMITIVES "CONCAT;CONVOLUTION;DECONVOLUTION;INNER_PRODUCT;MATMUL;REORDER;POOLING;REDUCTION")
set(ONEDNN_ENABLED_ISA "XEHP;XEHPG;XEHPC")
if(CMAKE_COMPILER_IS_GNUCXX)
ie_add_compiler_flags(-Wno-undef -Wno-missing-declarations)
if(SUGGEST_OVERRIDE_SUPPORTED)
@@ -35,6 +37,12 @@ if(ENABLE_ONEDNN_FOR_GPU)
INSTALL_DIR "${ONEDNN_INSTALL_DIR}"
PREFIX "${ONEDNN_PREFIX_DIR}"
EXCLUDE_FROM_ALL ON
CMAKE_CACHE_ARGS
# The arguments below requires list to be passed as argument
# which doesn't work properly when passed to CMAKE_ARGS.
# Thus we pass it via CMAKE_CACHE_ARGS
"-DDNNL_ENABLE_PRIMITIVE:STRING=${ONEDNN_ENABLED_PRIMITIVES}"
"-DDNNL_ENABLE_PRIMITIVE_GPU_ISA:STRING=${ONEDNN_ENABLED_ISA}"
CMAKE_ARGS
"-DCMAKE_CXX_COMPILER=${CMAKE_CXX_COMPILER}"
"-DCMAKE_C_COMPILER=${CMAKE_C_COMPILER}"
@@ -55,6 +63,7 @@ if(ENABLE_ONEDNN_FOR_GPU)
"-DCMAKE_BUILD_TYPE=${CMAKE_BUILD_TYPE}"
"-DDNNL_ENABLE_CONCURRENT_EXEC=ON"
"-DDNNL_ENABLE_PRIMITIVE_CACHE=OFF"
"-DDNNL_ENABLE_WORKLOAD=INFERENCE"
"-DDNNL_ENABLE_JIT_PROFILING=${BUILD_SHARED_LIBS}"
"-DDNNL_ENABLE_ITT_TASKS=${BUILD_SHARED_LIBS}"
"-DDNNL_BUILD_TESTS=OFF"