github/ROCm
1
1
Fork 0
mirror of https://github.com/ROCm/ROCm.git synced 2026-04-27 03:01:52 -04:00
Code Issues Packages Projects Releases Wiki Activity

[BACKEND] Fix axisInfo analysis for div ops (#1157)

Browse Source
This commit is contained in:
Keren Zhou
2023-02-06 18:25:23 -08:00
committed by GitHub
parent a13ddf08e2
commit 681d04cf2b
2 changed files with 25 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
test/Analysis/test-alignment.mlir
View File

@@ -78,18 +78,24 @@ func @div() {
%1 = arith.constant dense<1> : tensor<128xi32>
// CHECK-NEXT: Contiguity: [128] ; Divisibility: [1073741824] ; Constancy: [1] ; ConstantValue: [None]
%2 = arith.divsi %0, %1 : tensor<128xi32>
// CHECK-NEXT: Contiguity: [1] ; Divisibility: [1073741824] ; Constancy: [1] ; ConstantValue: [None]
// CHECK-NEXT: Contiguity: [1] ; Divisibility: [1] ; Constancy: [1] ; ConstantValue: [None]
%3 = arith.divui %1, %0 : tensor<128xi32>
// CHECK-NEXT: Contiguity: [1] ; Divisibility: [64] ; Constancy: [128] ; ConstantValue: [64]
%4 = arith.constant dense<64> : tensor<128xi32>
// CHECK-NEXT: Contiguity: [1] ; Divisibility: [16777216] ; Constancy: [64] ; ConstantValue: [None]
%5 = arith.divsi %0, %4 : tensor<128xi32>
// CHECK-NEXT: Contiguity: [1] ; Divisibility: [16777216] ; Constancy: [1] ; ConstantValue: [None]
// CHECK-NEXT: Contiguity: [1] ; Divisibility: [1] ; Constancy: [1] ; ConstantValue: [None]
%6 = arith.divsi %4, %0 : tensor<128xi32>
// CHECK-NEXT: Contiguity: [1] ; Divisibility: [64] ; Constancy: [128] ; ConstantValue: [64]
%7 = arith.divsi %4, %1 : tensor<128xi32>
// CHECK-NEXT: Contiguity: [1] ; Divisibility: [2] ; Constancy: [128] ; ConstantValue: [66]
%7 = arith.constant dense<66> : tensor<128xi32>
// CHECK-NEXT: Contiguity: [1] ; Divisibility: [536870912] ; Constancy: [2] ; ConstantValue: [None]
%8 = arith.divui %0, %7 : tensor<128xi32>
%8 = arith.constant dense<66> : tensor<128xi32>
// CHECK-NEXT: Contiguity: [1] ; Divisibility: [1] ; Constancy: [2] ; ConstantValue: [None]
%9 = arith.divui %0, %8 : tensor<128xi32>
// CHECK-NEXT: Contiguity: [128] ; Divisibility: [8192] ; Constancy: [1] ; ConstantValue: [None]
%10 = tt.make_range {end = 8320 : i32, start = 8192 : i32} : tensor<128xi32>
// CHECK-NEXT: Contiguity: [1] ; Divisibility: [128] ; Constancy: [64] ; ConstantValue: [None]
%11 = arith.divsi %10, %4 : tensor<128xi32>
return
}