Searched refs:kernel_module (Results 1 – 4 of 4) sorted by relevance
5 // CHECK-LABEL: @kernel_module6 gpu.module @kernel_module {25 gpu.launch_func @kernel_module::@kernel38 gpu.module @kernel_module {49 gpu.launch_func @kernel_module::@kernel62 gpu.module @kernel_module {75 gpu.launch_func @kernel_module::@kernel88 gpu.module @kernel_module {101 gpu.launch_func @kernel_module::@kernel114 gpu.module @kernel_module {[all …]
156 auto kernel_module = SymbolTable::lookupNearestSymbolFrom<gpu::GPUModuleOp>( in matchAndRewrite() local158 assert(kernel_module && "expected a kernel module"); in matchAndRewrite()161 kernel_module->getAttrOfType<StringAttr>(gpu_binary_annotation_); in matchAndRewrite()163 kernel_module.emitOpError() in matchAndRewrite()169 SmallString<128> name_buffer(kernel_module.getName()); in matchAndRewrite()183 (kernel_module.getName() + "_" + kernel_name + "_kernel_name") in matchAndRewrite()
6 // CHECK-NOT: gpu.module @kernel_module7 gpu.module @kernel_module attributes {gpu.binary_blob = "BLOB!"} {31 gpu.launch_func @kernel_module::@the_kernel
10 gpu.module @kernel_module attributes {23 gpu.launch_func @kernel_module::@kernel