[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:
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 } });
|
||||
|
||||
@@ -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"
|
||||
|
||||
Reference in new issue
Block a user