| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/gemm/ |
| H A D | gemm_driver.cpp | 405 if (is_int8) { in gemm_kernel()
579 if (is_int8) { in gemm_kernel_driver()
607 if (is_int8) { in gemm_kernel_driver()
614 if (is_int8) in gemm_kernel_driver()
821 if (is_int8) { in kernel_driver_parallel_acopiedbcopy()
827 if (is_int8) in kernel_driver_parallel_acopiedbcopy()
840 if (is_int8) in kernel_driver_parallel_acopiedbcopy()
1406 if (is_int8) { in parallel_a_copy()
1419 if (is_int8) in parallel_a_copy()
1689 pack_dst->setup(nthr_goal, do_a && is_int8, !do_a && is_int8); in gemm_threading_driver()
[all …]
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/gemm/ |
| H A D | gemm_driver.cpp | 411 if (is_int8) { in gemm_kernel()
579 if (is_int8) { in gemm_kernel_driver()
607 if (is_int8) { in gemm_kernel_driver()
614 if (is_int8) in gemm_kernel_driver()
821 if (is_int8) { in kernel_driver_parallel_acopiedbcopy()
827 if (is_int8) in kernel_driver_parallel_acopiedbcopy()
840 if (is_int8) in kernel_driver_parallel_acopiedbcopy()
1406 if (is_int8) { in parallel_a_copy()
1419 if (is_int8) in parallel_a_copy()
1689 pack_dst->setup(nthr_goal, do_a && is_int8, !do_a && is_int8); in gemm_threading_driver()
[all …]
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/rnn/ |
| H A D | rnn_brgemm_utils.cpp | 33 bool is_int8, bool is_bf16);
35 dim_t K1, dim_t K2, bool is_int8);
76 if (is_int8) { in brgemm_calc_isa()
88 bool is_int8, bool is_bf16) { in brgemm_calc_k_block() argument
93 return brgemm_calc_k_block_amx(K1, K2, is_int8); in brgemm_calc_k_block()
102 dim_t K1, dim_t K2, bool is_int8) { in brgemm_calc_k_block_amx() argument
316 const auto is_int8 = rnn.is_int8(); in configure_brgemm() local
319 const dim_t padding = (is_int8 ? 4 : (is_bf16 ? 2 : 1)); in configure_brgemm()
686 const auto is_int8 = rnn.is_int8(); in configure_brgemm() local
704 const bool is_int8_amx = rnn.is_int8() in configure_brgemm()
[all …]
|
| /dports/math/onednn/oneDNN-2.5.1/src/gpu/jit/ |
| H A D | xe_hp_convolution.hpp | 87 bool is_int8 = (acc_dt == s32); in data_types_ok() local
88 is_int8 &= utils::one_of(src_dt, u8, s8); in data_types_ok()
89 is_int8 &= utils::one_of(wei_dt, u8, s8); in data_types_ok()
90 is_int8 &= utils::one_of(dst_dt, u8, s8, s32, f32); in data_types_ok()
91 if (is_int8) return true; in data_types_ok()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/brgemm/ |
| H A D | brgemm.cpp | 150 brg->embd_bcst = !brg->is_int8 && !brg->is_bf16 in brgemm_blocking()
385 brg->is_int8 = (one_of(brg->dt_a, data_type::u8, data_type::s8) in brgemm_desc_init()
390 if (!brg->is_int8 && !brg->is_bf16 && !brg->is_f32) in brgemm_desc_init()
392 brg->dt_c = (brg->is_int8) ? data_type::s32 : data_type::f32; in brgemm_desc_init()
400 if (!IMPLICATION(brg->is_int8, mayiuse(avx512_core_vnni))) in brgemm_desc_init()
409 if (brg->is_int8 && isa == avx512_core_bf16_amx_int8) { in brgemm_desc_init()
420 brg->is_int8_amx = brg->is_int8 && mayiuse(avx512_core_bf16_amx_int8); in brgemm_desc_init()
488 brg->is_int8 = one_of(brg->dt_a, data_type::u8, data_type::s8) in brdgmm_desc_init()
494 if (!brg->is_int8 && !brg->is_bf16 && !brg->is_f32) in brdgmm_desc_init()
496 brg->dt_c = (brg->is_int8) ? data_type::s32 : data_type::f32; in brdgmm_desc_init()
[all …]
|
| /dports/math/onednn/oneDNN-2.5.1/src/gpu/ocl/ |
| H A D | xe_hpc_1x1_convolution.cpp | 40 bool is_int8 = src_mdw.data_type() == data_type::u8 in init_conf() local
47 conf.calc_block = (is_int8) ? 16 : 32; in init_conf()
48 conf.oc_block = (is_int8) ? 32 : 16; in init_conf()
49 conf.ic_block = (is_int8) ? 32 : 16; in init_conf()
91 if (is_int8) { in init_conf()
|
| H A D | xe_hp_1x1_convolution.cpp | 44 bool is_int8 = src_mdw.data_type() == data_type::u8 in init_conf() local
52 conf.oc_block = (is_int8) ? 32 : 16; in init_conf()
53 conf.ic_block = (is_int8) ? 32 : 16; in init_conf()
130 if (is_int8) { in init_conf()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ocl/gemm/ |
| H A D | gemm_with_post_ops.cpp | 113 bool is_int8 = src_md(1)->data_type == data_type::s8; in init_kernel_ctx() local
114 kernel_ctx.set_data_type(is_int8 ? data_type::f32 : c_type); in init_kernel_ctx()
118 kernel_ctx, is_int8 ? data_type::f32 : desc()->acc_type, "ACC"); in init_kernel_ctx()
121 : is_int8 ? data_type::f32 : c_type, in init_kernel_ctx()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/rnn/ |
| H A D | rnn_brgemm_utils.cpp | 32 dim_t l2_cache_size, bool is_int8, bool is_bf16);
34 dim_t K1, dim_t K2, bool is_int8);
55 dim_t l2_cache_size, bool is_int8, bool is_bf16) { in brgemm_calc_k_block_with_isa() argument
57 = is_int8 && x64::mayiuse(x64::avx512_core_bf16_amx_int8); in brgemm_calc_k_block_with_isa()
65 if (is_int8) { isa = x64::avx512_core_vnni; } in brgemm_calc_k_block_with_isa()
69 const auto result = brgemm_calc_k_block_amx(K1, K2, is_int8); in brgemm_calc_k_block_with_isa()
96 dim_t K1, dim_t K2, bool is_int8) { in brgemm_calc_k_block_amx() argument
299 l2_cache_size, rnn.is_int8(), rnn.is_bf16()); in configure_brgemm()
628 rnn.is_int8(), rnn.is_bf16()); in configure_brgemm()
634 const bool is_int8_amx = rnn.is_int8() in configure_brgemm()
[all …]
|
| /dports/devel/py-PeachPy/PeachPy-01d1515/peachpy/ |
| H A D | util.py | 9 if is_int8(n):
59 def is_int8(n): function
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/brgemm/ |
| H A D | brgemm.cpp | 166 brg->is_int8 = (one_of(brg->dt_a, data_type::u8, data_type::s8) in brgemm_desc_init()
171 if (!brg->is_int8 && !brg->is_bf16 && !brg->is_f32) in brgemm_desc_init()
173 brg->dt_c = (brg->is_int8) ? data_type::s32 : data_type::f32; in brgemm_desc_init()
181 if (!IMPLICATION(brg->is_int8, mayiuse(avx512_core_vnni))) in brgemm_desc_init()
190 if (brg->is_int8 && isa == avx512_core_bf16_amx_int8) { in brgemm_desc_init()
201 brg->is_int8_amx = brg->is_int8 && mayiuse(avx512_core_bf16_amx_int8); in brgemm_desc_init()
206 = brg->is_int8 && !brg->is_int8_amx && brg->dt_a == data_type::s8; in brgemm_desc_init()
248 brg->embd_bcst = !brg->is_int8 && !brg->is_bf16 in brgemm_desc_init()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/aarch64/ |
| H A D | acl_eltwise_utils.cpp | 41 const bool is_int8 = one_of(ed.data_desc.data_type, s8, u8); in acl_eltwise_check() local
91 if (is_int8) { in acl_eltwise_check()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/benchdnn/rnn/ |
| H A D | rnn_aux.cpp | 373 if (!prb.cfg.is_int8()) return in; in maybe_deq()
380 if (!prb.cfg.is_int8()) return in; in maybe_deq()
389 if (!prb.cfg.is_int8()) return in; in maybe_deq()
400 if (!prb.cfg.is_int8()) return h; in maybe_q()
635 if (prb.is_int8() && kind == DST_ITER_C) check_norm0 = false; in compare_dat()
728 if (!cfg.is_int8()) { in set_qparams()
781 int64_t divisor = next_pow2((acc_dim + n_cscale) * (is_int8() ? 2 : 1)); in set_tparams()
|
| H A D | ref_rnn_fwd.cpp | 136 if (prb.is_int8()) in copy_init_fwd()
146 if (prb.is_int8()) in copy_init_fwd()
177 prb.is_int8()); in copy_res_fwd()
265 if (prb.is_int8()) { in rnn_linear_fwd()
|
| /dports/math/onednn/oneDNN-2.5.1/tests/benchdnn/rnn/ |
| H A D | rnn_aux.cpp | 373 if (!prb.cfg.is_int8()) return in; in maybe_deq()
380 if (!prb.cfg.is_int8()) return in; in maybe_deq()
389 if (!prb.cfg.is_int8()) return in; in maybe_deq()
400 if (!prb.cfg.is_int8()) return h; in maybe_q()
635 if (prb.is_int8() && kind == DST_ITER_C) check_norm0 = false; in compare_dat()
728 if (!cfg.is_int8()) { in set_qparams()
781 int64_t divisor = next_pow2((acc_dim + n_cscale) * (is_int8() ? 2 : 1)); in set_tparams()
|
| H A D | ref_rnn_fwd.cpp | 136 if (prb.is_int8()) in copy_init_fwd()
146 if (prb.is_int8()) in copy_init_fwd()
177 prb.is_int8()); in copy_res_fwd()
265 if (prb.is_int8()) { in rnn_linear_fwd()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/ |
| H A D | jit_brdgmm_dw_conv.cpp | 82 const bool is_int8 = one_of(src_type, u8) && wei_type == s8 in init() local
87 : (is_int8 ? avx512_core_vnni : avx512_core_bf16); in init()
90 if (is_int8) skip_mask |= skip_mask_t::oscale; in init()
93 && one_of(true, is_f32, is_int8, is_bf16) && mayiuse(isa) in init()
149 = (cd.prop_kind == prop_kind::forward_inference || is_int8); in init()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/matmul/ |
| H A D | brgemm_matmul_utils.hpp | 217 inline bool is_int8() const { return int8_dt; } in is_int8() function
220 return this->is_int8() && bgmmc.dst_dt == data_type::bf16; in is_int8_with_bf16_dst()
|
| /dports/www/qt5-webengine/qtwebengine-everywhere-src-5.15.2/src/3rdparty/chromium/tools/traceline/traceline/ |
| H A D | assembler.h | 220 } else if (is_int8(disp)) { in Init()
243 } else if (is_int8(disp)) { in Init()
304 bool is_int8(int x) { return x >= -128 && x <= 127; } in is_int8() function
|
| /dports/www/chromium-legacy/chromium-88.0.4324.182/tools/traceline/traceline/ |
| H A D | assembler.h | 220 } else if (is_int8(disp)) { in Init()
243 } else if (is_int8(disp)) { in Init()
304 bool is_int8(int x) { return x >= -128 && x <= 127; } in is_int8() function
|
| /dports/math/octave/octave-6.4.0/libinterp/corefcn/ |
| H A D | mxarray.in.h | 178 virtual int is_int8 (void) const = 0;
401 int is_int8 (void) const { return rep->is_int8 (); } in is_int8() function
|
| /dports/www/qt5-webengine/qtwebengine-everywhere-src-5.15.2/src/3rdparty/chromium/v8/src/codegen/x64/ |
| H A D | assembler-x64.cc | 146 } else if (is_int8(disp)) { in OperandBuilder()
163 } else if (is_int8(disp)) { in OperandBuilder()
255 DCHECK(is_int8(disp)); in set_disp8()
496 if (is_int8(disp)) { in FinalizeJumpOptimizationInfo()
572 CHECK(is_int8(disp)); in bind_to()
589 CHECK(is_int8(disp)); in bind_to()
813 if (is_int8(src.value_)) { in immediate_arithmetic_op_16()
832 if (is_int8(src.value_)) { in immediate_arithmetic_op_16()
1384 if (is_int8(imm.value_)) { in emit_imul()
1398 if (is_int8(imm.value_)) { in emit_imul()
[all …]
|
| /dports/www/node10/node-v10.24.1/deps/v8/src/x64/ |
| H A D | assembler-x64.cc | 171 } else if (is_int8(disp)) { in OperandBuilder()
188 } else if (is_int8(disp)) { in OperandBuilder()
279 DCHECK(is_int8(disp)); in set_disp8()
416 if (is_int8(disp)) { in GetCode()
488 CHECK(is_int8(disp)); in bind_to()
505 CHECK(is_int8(disp)); in bind_to()
751 if (is_int8(src.value_)) { in immediate_arithmetic_op_16()
770 if (is_int8(src.value_)) { in immediate_arithmetic_op_16()
1297 if (is_int8(imm.value_)) { in emit_imul()
1311 if (is_int8(imm.value_)) { in emit_imul()
[all …]
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/rnn/ |
| H A D | rnn_utils.hpp | 332 inline bool is_int8() const { in is_int8() function
345 return brgemm_isa == x64::avx512_core_bf16_amx_int8 && is_int8(); in is_int8_amx()
711 || rnn.is_int8()) in init_conf()
724 rnn.copy_bias = rnn.is_int8(); in init_conf()
731 || rnn.is_int8() || is_bf16) in init_conf()
738 || rnn.is_int8() || is_bf16) in init_conf()
745 || rnn.is_int8() || is_bf16) in init_conf()
801 const bool need_compensation = rnn.is_int8(); in init_conf()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/rnn/ |
| H A D | rnn_utils.hpp | 342 inline bool is_int8() const { in is_int8() function
355 return brgemm_isa == x64::avx512_core_bf16_amx_int8 && is_int8(); in is_int8_amx()
731 || rnn.is_int8()) in init_conf()
744 rnn.copy_bias = rnn.is_int8(); in init_conf()
751 || rnn.is_int8() || is_bf16) in init_conf()
758 || rnn.is_int8() || is_bf16) in init_conf()
765 || rnn.is_int8() || is_bf16) in init_conf()
838 const bool need_compensation = rnn.is_int8(); in init_conf()
|