Searched refs:jit_kernel (Results 1 – 3 of 3) sorted by relevance
| /dports/math/onednn/oneDNN-2.5.1/src/gpu/jit/conv/ |
| H A D | gen_convolution.cpp | 67 std::unique_ptr<jit::jit_generator_base> jit_kernel; in make_kernel() local
70 jit_kernel = make_generator<KernelT, gpu_gen9>( in make_kernel()
74 jit_kernel = make_generator<KernelT, gpu_xe_lp>( in make_kernel()
78 jit_kernel = make_generator<KernelT, gpu_xe_hp>( in make_kernel()
82 jit_kernel = make_generator<KernelT, gpu_xe_hpg>( in make_kernel()
86 jit_kernel = make_generator<KernelT, gpu_xe_hpc>( in make_kernel()
92 if (!jit_kernel) return compute::kernel_t(); in make_kernel()
95 status_t status = primitive->create_kernel(engine, &kernel, *jit_kernel); in make_kernel()
|
| /dports/science/cp2k/cp2k-2e995eec7fd208c8a72d9544807bd8b8ba8cd1cc/exts/dbcsr/src/acc/libsmm_acc/ |
| H A D | libsmm_acc.cpp | 99 inline void jit_kernel(ACC_DRV(function)& kern_func, libsmm_acc_algo algo, int tile_m, int tile_n, … in jit_kernel() function
208 …jit_kernel(kern_func, algo, tile_m, tile_n, w, v, threads, grouping, minblocks, h_mnk[0], h_mnk[1]… in add_kernel_handle_to_jitted_kernels()
|
| /dports/math/dbcsr/dbcsr-2.1.0/src/acc/libsmm_acc/ |
| H A D | libsmm_acc.cpp | 94 inline void jit_kernel(ACC_DRV(function)& kern_func, libsmm_acc_algo algo, int tile_m, int tile_n, … in jit_kernel() function
210 …jit_kernel(kern_func, algo, tile_m, tile_n, w, v, threads, grouping, minblocks, h_mnk[0], h_mnk[1]… in add_kernel_handle_to_jitted_kernels()
|