Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/test/Dialect/Linalg/
H A Dtile-and-distribute.mlir74 // CHECK-DAG: %[[NBLOCKSY:.*]] = gpu.grid_dim y
76 // CHECK-DAG: %[[NBLOCKSX:.*]] = gpu.grid_dim x
133 // CHECK-DAG: %[[NBLOCKSX:.*]] = gpu.grid_dim x
163 // CHECK-DAG: %[[NBLOCKSY:.*]] = gpu.grid_dim y
190 // CHECK-DAG: %[[NBLOCKSY:.*]] = gpu.grid_dim y
192 // CHECK-DAG: %[[NBLOCKSX:.*]] = gpu.grid_dim x
H A Dtile-fuse-and-distribute.mlir14 // CHECK-DAG: %[[NBLOCKSY:.*]] = gpu.grid_dim y
16 // CHECK-DAG: %[[NBLOCKSX:.*]] = gpu.grid_dim x
/llvm-project-15.0.7/mlir/test/Conversion/GPUToSPIRV/
H A Dload-store.mlir52 %6 = gpu.grid_dim x
53 %7 = gpu.grid_dim y
54 %8 = gpu.grid_dim z
H A Dbuiltins.mlir186 %0 = gpu.grid_dim x
/llvm-project-15.0.7/mlir/test/Dialect/GPU/
H A Dint-range-interface.mlir55 %grid_dim_x = gpu.grid_dim x
56 %grid_dim_y = gpu.grid_dim y
57 %grid_dim_z = gpu.grid_dim z
H A Dops.mlir73 %gDimX = gpu.grid_dim x
74 %gDimY = gpu.grid_dim y
75 %gDimZ = gpu.grid_dim z
H A Doutlining.mlir50 // CHECK-NEXT: = gpu.grid_dim x
51 // CHECK-NEXT: = gpu.grid_dim y
52 // CHECK-NEXT: = gpu.grid_dim z
/llvm-project-15.0.7/mlir/test/Conversion/GPUToROCDL/
H A Dgpu-to-rocdl.mlir44 %gDimX = gpu.grid_dim x
47 %gDimY = gpu.grid_dim y
50 %gDimZ = gpu.grid_dim z
/llvm-project-15.0.7/polly/lib/External/ppcg/
H A Dgpu.h392 int grid_dim[2]; member
H A Dgpu.c653 kernel->grid_dim[0] = 32768; in read_grid_sizes()
656 kernel->grid_dim[0] = 256; in read_grid_sizes()
657 kernel->grid_dim[1] = 256; in read_grid_sizes()
662 read_sizes_from_set(size, kernel->grid_dim, &kernel->n_grid); in read_grid_sizes()
678 kernel->grid_dim, kernel->n_grid); in read_grid_and_block_sizes()
3900 kernel->grid_dim); in gpu_create_kernel()
3904 node = snap_band_to_sizes(node, kernel->grid_dim, in gpu_create_kernel()
/llvm-project-15.0.7/mlir/test/Conversion/GPUToNVVM/
H A Dgpu-to-nvvm.mlir45 %gDimX = gpu.grid_dim x
48 %gDimY = gpu.grid_dim y
51 %gDimZ = gpu.grid_dim z
/llvm-project-15.0.7/mlir/include/mlir/Dialect/GPU/IR/
H A DGPUOps.td77 def GPU_GridDimOp : GPU_IndexOp<"grid_dim"> {
85 %gDimZ = gpu.grid_dim z
401 %gDimX = gpu.grid_dim x
402 %gDimY = gpu.grid_dim y
403 %gDimZ = gpu.grid_dim z