Lines Matching defs:gpu
4 import mlir.dialects.gpu as gpu
5 import mlir.dialects.gpu.passes
20 PassManager.parse("any(gpu-kernel-outlining)")
29 gpu.BlockDimOp(gpu.Dimension.y)
30 # CHECK: %block_dim_y = gpu.block_dim y
39 format = gpu.CompilationTarget.Fatbin
42 o = gpu.ObjectAttr.get(target, format, object, properties)
43 # CHECK: #gpu.object<#nvvm.target, properties = {O = 2 : i32}, "BC\C0\DE5\14\00\00\05\00\00\00b\0C0$MY\BEf">
47 o = gpu.ObjectAttr.get(target, format, object)
48 # CHECK: #gpu.object<#nvvm.target, "BC\C0\DE5\14\00\00\05\00\00\00b\0C0$MY\BEf">
54 o = gpu.ObjectAttr.get(target, format, object)
55 # CHECK: #gpu.object<#nvvm.target, "//\0A// Generated by LLVM NVPTX Back-End\0A//\0A\0A.version 6.0\0A.target sm_50">
61 '#gpu.kernel_table<[#gpu.kernel_metadata<"kernel", () -> ()>]>'
63 o = gpu.ObjectAttr.get(target, format, object, kernels=kernelTable)
64 # CHECK: #gpu.object<#nvvm.target, kernels = <[#gpu.kernel_metadata<"kernel", () -> ()>]>, "BC\C0\DE5\14\00\00\05\00\00\00b\0C0$MY\BEf">