Home
last modified time | relevance | path

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

/llvm-project/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], kernel_func.type.results),
84 num_results = len(kernel_func.type.results)
95 kernel_func,
H A Dbenchmark_sparse.py53 kernel_func = get_kernel_func_from_module(module)
55 wrapped_func = emit_benchmark_wrapped_main_func(kernel_func, timer_func)
57 str(timer_func) + str(wrapped_func) + str(kernel_func)
/llvm-project/mlir/test/Target/LLVMIR/
H A Dnvvmir.mlir581 llvm.func @kernel_func() attributes {nvvm.kernel} {
585 // CHECK: ptx_kernel void @kernel_func
589 llvm.func @kernel_func() attributes {nvvm.kernel, nvvm.maxntid = array<i32: 1, 23, 32>} {
593 // CHECK: define ptx_kernel void @kernel_func
595 // CHECK: {ptr @kernel_func, !"maxntidx", i32 1}
596 // CHECK: {ptr @kernel_func, !"maxntidy", i32 23}
597 // CHECK: {ptr @kernel_func, !"maxntidz", i32 32}
600 llvm.func @kernel_func() attributes {nvvm.kernel, nvvm.reqntid = array<i32: 1, 23, 32>} {
604 // CHECK: define ptx_kernel void @kernel_func
606 // CHECK: {ptr @kernel_func, !"reqntid
[all...]
H A Drocdl.mlir38 llvm.func @kernel_func() attributes {rocdl.kernel} {
39 // CHECK-LABEL: amdgpu_kernel void @kernel_func()
/llvm-project/llvm/test/CodeGen/NVPTX/
H A Dsimple-call.ll12 ; CHECK: .entry kernel_func
13 define ptx_kernel void @kernel_func(ptr %a) {
H A Dcall-with-alloca-buffer.ll9 ; extern "C" __attribute__((global)) void kernel_func(ptr a) {
18 ; CHECK: .visible .entry kernel_func
19 define ptx_kernel void @kernel_func(ptr %a) {
H A Dunreachable.ll24 define ptx_kernel void @kernel_func() {
25 ; NO-TRAP-UNREACHABLE-LABEL: kernel_func(
40 ; NO-TRAP-AFTER-NORETURN-LABEL: kernel_func(
56 ; TRAP-LABEL: kernel_func(
69 ; BUG-FIXED-LABEL: kernel_func(
H A Dshuffle-vec-undef-init.ll6 define void @kernel_func(ptr %in.vec, ptr %out.vec0) nounwind {
7 ; CHECK-LABEL: kernel_func(
H A Dcalling-conv.ll9 ; CHECK: .entry kernel_func
10 define ptx_kernel void @kernel_func(ptr %a) {
/llvm-project/mlir/test/Dialect/LLVMIR/
H A Dnvvm.mlir535 llvm.func @kernel_func(%arg0: !llvm.ptr {llvm.byval = i32, nvvm.grid_constant}) attributes {nvvm.kernel} {
542 llvm.func @kernel_func(%arg0: !llvm.ptr {llvm.byval = i32, nvvm.grid_constant}) {
549 llvm.func @kernel_func(%arg0: !llvm.ptr {nvvm.grid_constant}) attributes {nvvm.kernel} {
556 llvm.func @kernel_func(%arg0: !llvm.ptr {llvm.byval = i32, nvvm.grid_constant = true}) attributes {nvvm.kernel} {
/llvm-project/mlir/test/Conversion/GPUToROCDL/
H A Dgpu-to-rocdl.mlir572 // CHECK-LABEL: @kernel_func
576 gpu.func @kernel_func() kernel {
/llvm-project/mlir/test/Conversion/GPUToNVVM/
H A Dgpu-to-nvvm.mlir589 // CHECK-LABEL: @kernel_func
593 gpu.func @kernel_func() kernel {