/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/benchdnn/ |
H A D | dnnl_common.cpp | 100 DNN_SAFE_V(dnnl_set_cpu_isa_hints(dnnl_cpu_isa_no_hints)); in init_isa_settings() 102 DNN_SAFE_V(dnnl_set_cpu_isa_hints(dnnl_cpu_isa_prefer_ymm)); in init_isa_settings() 192 DNN_SAFE_V(dnnl_engine_get_kind(engine, &engine_kind)); in get_engine_kind() 721 DNN_SAFE_V(dnnl_engine_create(&engine_, engine_kind, idx)); in engine_t() 735 DNN_SAFE_V(dnnl_engine_get_kind(other.engine_, &engine_kind)); in engine_t() 743 DNN_SAFE_V(dnnl_sycl_interop_engine_create(&engine_, dev, ctx)); in engine_t() 745 DNN_SAFE_V(dnnl_engine_create(&engine_, dnnl_cpu, 0)); in engine_t() 751 DNN_SAFE_V(dnnl_ocl_interop_get_device(other.engine_, &dev)); in engine_t() 753 DNN_SAFE_V(dnnl_ocl_interop_engine_create(&engine_, dev, ctx)); in engine_t() 767 if (is_owner_) DNN_SAFE_V(dnnl_engine_destroy(engine_)); in ~engine_t() [all …]
|
H A D | dnnl_common.hpp | 55 #define DNN_SAFE_V(f) \ macro 254 DNN_SAFE_V(dnnl_primitive_destroy(t)); in destroy() 261 DNN_SAFE_V(dnnl_primitive_desc_destroy(t)); in destroy() 268 DNN_SAFE_V(dnnl_primitive_attr_destroy(t)); in destroy()
|
H A D | dnn_types.cpp | 808 DNN_SAFE_V(dnnl_primitive_attr_create(&dnnl_attr)); in create_dnnl_attr() 818 DNN_SAFE_V(dnnl_primitive_attr_set_output_scales( in create_dnnl_attr() 832 DNN_SAFE_V(dnnl_primitive_attr_set_scales( in create_dnnl_attr() 850 DNN_SAFE_V(dnnl_primitive_attr_set_zero_points( in create_dnnl_attr() 857 DNN_SAFE_V(dnnl_post_ops_create(&ops)); in create_dnnl_attr() 863 DNN_SAFE_V(dnnl_post_ops_append_sum_v2( in create_dnnl_attr() 880 DNN_SAFE_V(dnnl_post_ops_append_dw(ops, wei_dt, bia_dt, in create_dnnl_attr() 889 DNN_SAFE_V(dnnl_post_ops_append_binary( in create_dnnl_attr() 895 DNN_SAFE_V(dnnl_primitive_attr_set_post_ops(dnnl_attr, ops)); in create_dnnl_attr() 901 DNN_SAFE_V(dnnl_post_ops_destroy(ops)); in create_dnnl_attr() [all …]
|
H A D | dnnl_memory.hpp | 191 DNN_SAFE_V(dnnl_memory_map_data(m_, &mapped_ptr_)); in map() 199 DNN_SAFE_V(dnnl_memory_unmap_data(m_, mapped_ptr_)); in unmap()
|
/dports/math/onednn/oneDNN-2.5.1/tests/benchdnn/ |
H A D | dnnl_common.cpp | 100 DNN_SAFE_V(dnnl_set_cpu_isa_hints(dnnl_cpu_isa_no_hints)); in init_isa_settings() 102 DNN_SAFE_V(dnnl_set_cpu_isa_hints(dnnl_cpu_isa_prefer_ymm)); in init_isa_settings() 200 DNN_SAFE_V(dnnl_engine_get_kind(engine, &engine_kind)); in get_engine_kind() 800 DNN_SAFE_V(dnnl_engine_create(&engine_, engine_kind, idx)); in engine_t() 814 DNN_SAFE_V(dnnl_engine_get_kind(other.engine_, &engine_kind)); in engine_t() 822 DNN_SAFE_V(dnnl_sycl_interop_engine_create(&engine_, dev, ctx)); in engine_t() 824 DNN_SAFE_V(dnnl_engine_create(&engine_, dnnl_cpu, 0)); in engine_t() 830 DNN_SAFE_V(dnnl_ocl_interop_get_device(other.engine_, &dev)); in engine_t() 832 DNN_SAFE_V(dnnl_ocl_interop_engine_create(&engine_, dev, ctx)); in engine_t() 846 if (is_owner_) DNN_SAFE_V(dnnl_engine_destroy(engine_)); in ~engine_t() [all …]
|
H A D | dnn_types.cpp | 883 DNN_SAFE_V(dnnl_primitive_attr_create(&dnnl_attr)); in create_dnnl_attr() 893 DNN_SAFE_V(dnnl_primitive_attr_set_output_scales( in create_dnnl_attr() 907 DNN_SAFE_V(dnnl_primitive_attr_set_scales( in create_dnnl_attr() 925 DNN_SAFE_V(dnnl_primitive_attr_set_zero_points( in create_dnnl_attr() 932 DNN_SAFE_V(dnnl_post_ops_create(&ops)); in create_dnnl_attr() 938 DNN_SAFE_V(dnnl_post_ops_append_sum_v3( in create_dnnl_attr() 964 DNN_SAFE_V(dnnl_post_ops_append_binary( in create_dnnl_attr() 969 DNN_SAFE_V(dnnl_post_ops_append_prelu(ops, mask)); in create_dnnl_attr() 974 DNN_SAFE_V(dnnl_primitive_attr_set_post_ops(dnnl_attr, ops)); in create_dnnl_attr() 980 DNN_SAFE_V(dnnl_post_ops_destroy(ops)); in create_dnnl_attr() [all …]
|
H A D | dnnl_common.hpp | 56 #define DNN_SAFE_V(f) \ macro 259 DNN_SAFE_V(dnnl_primitive_destroy(t)); in destroy() 266 DNN_SAFE_V(dnnl_primitive_desc_destroy(t)); in destroy() 273 DNN_SAFE_V(dnnl_primitive_attr_destroy(t)); in destroy()
|
H A D | dnnl_memory.hpp | 191 DNN_SAFE_V(dnnl_memory_map_data(m_, &mapped_ptr_)); in map() 199 DNN_SAFE_V(dnnl_memory_unmap_data(m_, mapped_ptr_)); in unmap()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/benchdnn/rnn/ |
H A D | rnn.cpp | 102 DNN_SAFE_V(dnnl_primitive_attr_create(dnnl_attr)); in create_dnnl_rnn_attr() 105 DNN_SAFE_V(dnnl_primitive_attr_set_rnn_tparams(*dnnl_attr, true, in create_dnnl_rnn_attr() 108 DNN_SAFE_V(dnnl_primitive_attr_set_rnn_weights_qparams( in create_dnnl_rnn_attr() 112 DNN_SAFE_V(dnnl_primitive_attr_set_rnn_weights_projection_qparams( in create_dnnl_rnn_attr() 117 DNN_SAFE_V(dnnl_primitive_attr_set_rnn_data_qparams( in create_dnnl_rnn_attr() 120 DNN_SAFE_V(dnnl_primitive_attr_set_scratchpad_mode( in create_dnnl_rnn_attr()
|
H A D | lstm_cell.cpp | 256 DNN_SAFE_V(diff_dst == nullptr ? dnnl_out_of_memory : dnnl_success); in lstm_bwd()
|
/dports/math/onednn/oneDNN-2.5.1/tests/benchdnn/rnn/ |
H A D | rnn.cpp | 102 DNN_SAFE_V(dnnl_primitive_attr_create(dnnl_attr)); in create_dnnl_rnn_attr() 105 DNN_SAFE_V(dnnl_primitive_attr_set_rnn_tparams(*dnnl_attr, true, in create_dnnl_rnn_attr() 108 DNN_SAFE_V(dnnl_primitive_attr_set_rnn_weights_qparams( in create_dnnl_rnn_attr() 112 DNN_SAFE_V(dnnl_primitive_attr_set_rnn_weights_projection_qparams( in create_dnnl_rnn_attr() 117 DNN_SAFE_V(dnnl_primitive_attr_set_rnn_data_qparams( in create_dnnl_rnn_attr() 120 DNN_SAFE_V(dnnl_primitive_attr_set_scratchpad_mode( in create_dnnl_rnn_attr()
|
H A D | lstm_cell.cpp | 256 DNN_SAFE_V(diff_dst == nullptr ? dnnl_out_of_memory : dnnl_success); in lstm_bwd()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/benchdnn/conv/ |
H A D | ref_conv.cpp | 30 DNN_SAFE_V(dnnl_primitive_get_primitive_desc(prim_ref, &pd_ref)); in exec_conv() 31 DNN_SAFE_V(dnnl_primitive_desc_query( in exec_conv()
|