Searched refs:block_dim (Results 1 – 18 of 18) sorted by relevance
| /llvm-project-15.0.7/mlir/test/Dialect/GPU/ |
| H A D | promotion.mlir | 17 // CHECK-DAG: %[[bdx:.*]] = gpu.block_dim x 18 // CHECK-DAG: %[[bdy:.*]] = gpu.block_dim y 19 // CHECK-DAG: %[[bdz:.*]] = gpu.block_dim z 71 // CHECK-DAG: %[[bdx:.*]] = gpu.block_dim x 72 // CHECK-DAG: %[[bdy:.*]] = gpu.block_dim y 73 // CHECK-DAG: %[[bdz:.*]] = gpu.block_dim z
|
| H A D | int-range-interface.mlir | 77 %block_dim_x = gpu.block_dim x 78 %block_dim_y = gpu.block_dim y 79 %block_dim_z = gpu.block_dim z
|
| H A D | all-reduce.mlir | 19 // CHECK: [[VAL_11:%.*]] = gpu.block_dim x 21 // CHECK: [[VAL_13:%.*]] = gpu.block_dim y 23 // CHECK: [[VAL_15:%.*]] = gpu.block_dim z
|
| H A D | ops.mlir | 65 %bDimX = gpu.block_dim x 66 %bDimY = gpu.block_dim y 67 %bDimZ = gpu.block_dim z
|
| H A D | outlining.mlir | 53 // CHECK-NEXT: %[[BDIM:.*]] = gpu.block_dim x 54 // CHECK-NEXT: = gpu.block_dim y 55 // CHECK-NEXT: = gpu.block_dim z
|
| H A D | all-reduce-max.mlir | 19 // CHECK: [[VAL_11:%.*]] = gpu.block_dim x 21 // CHECK: [[VAL_13:%.*]] = gpu.block_dim y 23 // CHECK: [[VAL_15:%.*]] = gpu.block_dim z
|
| /llvm-project-15.0.7/mlir/test/Integration/GPU/ROCM/ |
| H A D | gpu-to-hsaco.mlir | 14 %block_dim = memref.dim %arg1, %c0 : memref<?xf32> 16 threads(%tx, %ty, %tz) in (%block_x = %block_dim, %block_y = %c1, %block_z = %c1) {
|
| H A D | vecadd.mlir | 15 %block_dim = memref.dim %arg0, %c0 : memref<?xf32> 17 threads(%tx, %ty, %tz) in (%block_x = %block_dim, %block_y = %c1, %block_z = %c1) {
|
| /llvm-project-15.0.7/mlir/test/Conversion/GPUToSPIRV/ |
| H A D | builtins.mlir | 94 %0 = gpu.block_dim x 116 %0 = gpu.block_dim y 138 %0 = gpu.block_dim z 242 %0 = gpu.block_dim x 266 %0 = gpu.block_dim y 290 %0 = gpu.block_dim z
|
| H A D | load-store.mlir | 55 %9 = gpu.block_dim x 56 %10 = gpu.block_dim y 57 %11 = gpu.block_dim z
|
| /llvm-project-15.0.7/mlir/test/Conversion/GPUToROCDL/ |
| H A D | gpu-to-rocdl.mlir | 24 %bDimX = gpu.block_dim x 27 %bDimY = gpu.block_dim y 30 %bDimZ = gpu.block_dim z
|
| /llvm-project-15.0.7/polly/lib/External/ppcg/ |
| H A D | gpu.h | 393 int block_dim[3]; member
|
| H A D | gpu.c | 624 kernel->block_dim[0] = 512; in read_block_sizes() 627 kernel->block_dim[0] = 32; in read_block_sizes() 628 kernel->block_dim[1] = 16; in read_block_sizes() 631 kernel->block_dim[0] = 32; in read_block_sizes() 632 kernel->block_dim[1] = 4; in read_block_sizes() 633 kernel->block_dim[2] = 4; in read_block_sizes() 676 kernel->block_dim, kernel->n_block); in read_grid_and_block_sizes() 1147 extract_fixed_size(block, kernel->block_dim); in extract_block_size() 3064 kernel->block_dim, kernel->thread_ids); in insert_context() 3557 kernel->block_dim); in add_copies_group_shared() [all …]
|
| H A D | cuda.c | 630 p = print_reverse_list(p, kernel->n_block, kernel->block_dim); in print_host_user()
|
| H A D | gpu_group.c | 1588 v = isl_val_int_from_si(ctx, kernel->block_dim[i]); in compute_privatization()
|
| /llvm-project-15.0.7/mlir/test/Conversion/GPUToNVVM/ |
| H A D | gpu-to-nvvm.mlir | 25 %bDimX = gpu.block_dim x 28 %bDimY = gpu.block_dim y 31 %bDimZ = gpu.block_dim z
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/GPU/IR/ |
| H A D | GPUOps.td | 53 def GPU_BlockDimOp : GPU_IndexOp<"block_dim"> { 61 %bDimX = gpu.block_dim x 393 %bDimX = gpu.block_dim x 394 %bDimY = gpu.block_dim y 395 %bDimZ = gpu.block_dim z
|
| /llvm-project-15.0.7/polly/lib/CodeGen/ |
| H A D | PPCGCodeGeneration.cpp | 1611 Value *Res = ConstantInt::get(Builder.getInt32Ty(), Kernel->block_dim[i]); in getBlockSizes()
|