113 lines · plain
1// RUN: mlir-opt %s -split-input-file | FileCheck %s2 3// -----4 5// Test with loop IVs.6func.func @test0(%arg0 : index, %arg1 : index) {7 %0 = memref.alloc() : memref<100x100xf32>8 %1 = memref.alloc() : memref<100x100xf32, affine_map<(d0, d1) -> (d0, d1)>, 2>9 %2 = memref.alloc() : memref<1xi32>10 %c0 = arith.constant 0 : index11 %c64 = arith.constant 64 : index12 affine.for %i0 = 0 to 10 {13 affine.for %i1 = 0 to 10 {14 affine.dma_start %0[%i0, %i1], %1[%i0, %i1], %2[%c0], %c6415 : memref<100x100xf32>, memref<100x100xf32, 2>, memref<1xi32>16 affine.dma_wait %2[%c0], %c64 : memref<1xi32>17// CHECK: affine.dma_start %{{.*}}[%{{.*}}, %{{.*}}], %{{.*}}[%{{.*}}, %{{.*}}], %{{.*}}[%{{.*}}], %{{.*}} : memref<100x100xf32>, memref<100x100xf32, 2>, memref<1xi32>18// CHECK: affine.dma_wait %{{.*}}[%{{.*}}], %{{.*}} : memref<1xi32>19 }20 }21 return22}23 24// -----25 26// Test with loop IVs and optional stride arguments.27func.func @test1(%arg0 : index, %arg1 : index) {28 %0 = memref.alloc() : memref<100x100xf32>29 %1 = memref.alloc() : memref<100x100xf32, affine_map<(d0, d1) -> (d0, d1)>, 2>30 %2 = memref.alloc() : memref<1xi32>31 %c0 = arith.constant 0 : index32 %c64 = arith.constant 64 : index33 %c128 = arith.constant 128 : index34 %c256 = arith.constant 256 : index35 affine.for %i0 = 0 to 10 {36 affine.for %i1 = 0 to 10 {37 affine.dma_start %0[%i0, %i1], %1[%i0, %i1], %2[%c0], %c64, %c128, %c25638 : memref<100x100xf32>, memref<100x100xf32, 2>, memref<1xi32>39 affine.dma_wait %2[%c0], %c64 : memref<1xi32>40// CHECK: affine.dma_start %{{.*}}[%{{.*}}, %{{.*}}], %{{.*}}[%{{.*}}, %{{.*}}], %{{.*}}[%{{.*}}], %{{.*}}, %{{.*}}, %{{.*}} : memref<100x100xf32>, memref<100x100xf32, 2>, memref<1xi32>41// CHECK: affine.dma_wait %{{.*}}[%{{.*}}], %{{.*}} : memref<1xi32>42 }43 }44 return45}46 47// -----48 49// Test with loop IVs and symbols (without symbol keyword).50func.func @test2(%arg0 : index, %arg1 : index) {51 %0 = memref.alloc() : memref<100x100xf32>52 %1 = memref.alloc() : memref<100x100xf32, affine_map<(d0, d1) -> (d0, d1)>, 2>53 %2 = memref.alloc() : memref<1xi32>54 %c0 = arith.constant 0 : index55 %c64 = arith.constant 64 : index56 affine.for %i0 = 0 to 10 {57 affine.for %i1 = 0 to 10 {58 affine.dma_start %0[%i0 + %arg0, %i1], %1[%i0, %i1 + %arg1 + 5],59 %2[%c0], %c6460 : memref<100x100xf32>, memref<100x100xf32, 2>, memref<1xi32>61 affine.dma_wait %2[%c0], %c64 : memref<1xi32>62// CHECK: affine.dma_start %{{.*}}[%{{.*}} + %{{.*}}, %{{.*}}], %{{.*}}[%{{.*}}, %{{.*}} + %{{.*}} + 5], %{{.*}}[%{{.*}}], %{{.*}} : memref<100x100xf32>, memref<100x100xf32, 2>, memref<1xi32>63// CHECK: affine.dma_wait %{{.*}}[%{{.*}}], %{{.*}} : memref<1xi32>64 }65 }66 return67}68 69// -----70 71// Test with loop IVs and symbols (with symbol keyword).72func.func @test3(%arg0 : index, %arg1 : index) {73 %0 = memref.alloc() : memref<100x100xf32>74 %1 = memref.alloc() : memref<100x100xf32, affine_map<(d0, d1) -> (d0, d1)>, 2>75 %2 = memref.alloc() : memref<1xi32>76 %c0 = arith.constant 0 : index77 %c64 = arith.constant 64 : index78 affine.for %i0 = 0 to 10 {79 affine.for %i1 = 0 to 10 {80 affine.dma_start %0[%i0 + symbol(%arg0), %i1],81 %1[%i0, %i1 + symbol(%arg1) + 7],82 %2[%i0 + %i1 + 11], %c6483 : memref<100x100xf32>, memref<100x100xf32, 2>, memref<1xi32>84 affine.dma_wait %2[%c0], %c64 : memref<1xi32>85// CHECK: affine.dma_start %{{.*}}[%{{.*}} + symbol(%{{.*}}), %{{.*}}], %{{.*}}[%{{.*}}, %{{.*}} + symbol(%{{.*}}) + 7], %{{.*}}[%{{.*}} + %{{.*}} + 11], %{{.*}} : memref<100x100xf32>, memref<100x100xf32, 2>, memref<1xi32>86// CHECK: affine.dma_wait %{{.*}}[%{{.*}}], %{{.*}} : memref<1xi32>87 }88 }89 return90}91 92// -----93 94// Test with loop IVs, symbols and constants in nested affine expressions.95func.func @test4(%arg0 : index, %arg1 : index) {96 %0 = memref.alloc() : memref<100x100xf32>97 %1 = memref.alloc() : memref<100x100xf32, 2>98 %2 = memref.alloc() : memref<1xi32>99 %c64 = arith.constant 64 : index100 affine.for %i0 = 0 to 10 {101 affine.for %i1 = 0 to 10 {102 affine.dma_start %0[(%i0 + symbol(%arg0)) floordiv 3, %i1],103 %1[%i0, (%i1 + symbol(%arg1)) mod 9 + 7],104 %2[%i0 + %i1 + 11], %c64105 : memref<100x100xf32>, memref<100x100xf32, 2>, memref<1xi32>106 affine.dma_wait %2[%i0 + %i1 + 11], %c64 : memref<1xi32>107// CHECK: affine.dma_start %{{.*}}[(%{{.*}} + symbol(%{{.*}})) floordiv 3, %{{.*}}], %{{.*}}[%{{.*}}, (%{{.*}} + symbol(%{{.*}})) mod 9 + 7], %{{.*}}[%{{.*}} + %{{.*}} + 11], %{{.*}} : memref<100x100xf32>, memref<100x100xf32, 2>, memref<1xi32>108// CHECK: affine.dma_wait %{{.*}}[%{{.*}} + %{{.*}} + 11], %{{.*}} : memref<1xi32>109 }110 }111 return112}113