Lines Matching refs:gpu

1 // RUN: mlir-opt %s -convert-gpu-to-nvvm='index-bitwidth=32' -split-input-file | FileCheck %s
5 gpu.module @test_module_0 {
11     %tIdX = gpu.thread_id x
12     %tIdY = gpu.thread_id y
13     %tIdZ = gpu.thread_id z
15     %bDimX = gpu.block_dim x
16     %bDimY = gpu.block_dim y
17     %bDimZ = gpu.block_dim z
19     %bIdX = gpu.block_id x
20     %bIdY = gpu.block_id y
21     %bIdZ = gpu.block_id z
23     %gDimX = gpu.grid_dim x
24     %gDimY = gpu.grid_dim y
25     %gDimZ = gpu.grid_dim z
28     %laneId = gpu.lane_id
41 gpu.module @test_module_1 {
53     %gpu_module = transform.structured.match ops{["gpu.module"]} in %toplevel_module
61       transform.apply_conversion_patterns.gpu.gpu_to_nvvm
62       transform.apply_conversion_patterns.gpu.gpu_wmma_to_nvvm
63       transform.apply_conversion_patterns.gpu.gpu_subgroup_reduce_to_nvvm {has_redux = true}
70       legal_ops = ["func.func", "gpu.module", "gpu.yield"],
71       illegal_dialects = ["gpu"],