Home
last modified time | relevance | path

Searched refs:kernel_func (Results 1 – 9 of 9) sorted by relevance

/llvm-project-15.0.7/mlir/benchmark/python/
H A Dcommon.py66 def emit_benchmark_wrapped_main_func(kernel_func, timer_func): argument
79 (kernel_func.arguments.types + [memref_of_i64_type],
80 kernel_func.type.results),
85 num_results = len(kernel_func.type.results)
96 kernel_func,
H A Dbenchmark_sparse.py52 kernel_func = get_kernel_func_from_module(module)
55 kernel_func,
59 str(timer_func) + str(wrapped_func) + str(kernel_func)
/llvm-project-15.0.7/llvm/test/CodeGen/NVPTX/
H A Dsimple-call.ll12 ; CHECK: .entry kernel_func
13 define void @kernel_func(float* %a) {
26 !1 = !{void (float*)* @kernel_func, !"kernel", i32 1}
H A Dcall-with-alloca-buffer.ll9 ; extern "C" __attribute__((global)) void kernel_func(float* a) {
18 ; CHECK: .visible .entry kernel_func
19 define void @kernel_func(float* %a) {
67 !0 = !{void (float*)* @kernel_func, !"kernel", i32 1}
H A Dcalling-conv.ll9 ; CHECK: .entry kernel_func
10 define ptx_kernel void @kernel_func(float* %a) {
/llvm-project-15.0.7/mlir/test/Target/LLVMIR/
H A Drocdl.mlir32 llvm.func @kernel_func() attributes {rocdl.kernel} {
33 // CHECK-LABEL: amdgpu_kernel void @kernel_func()
H A Dnvvmir.mlir345 llvm.func @kernel_func() attributes {nvvm.kernel} {
351 // CHECK: {ptr @kernel_func, !"kernel", i32 1}
/llvm-project-15.0.7/mlir/test/Conversion/GPUToROCDL/
H A Dgpu-to-rocdl.mlir381 // CHECK-LABEL: @kernel_func
385 gpu.func @kernel_func() kernel {
/llvm-project-15.0.7/mlir/test/Conversion/GPUToNVVM/
H A Dgpu-to-nvvm.mlir482 // CHECK-LABEL: @kernel_func
486 gpu.func @kernel_func() kernel {