Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
27 commits
Select commit Hold shift + click to select a range
c7fe682
Improve axis analysis to handle tt.make_tensor_ptr
etiotto Oct 9, 2024
ad3888f
Merge branch 'main' into etiotto/axis_analysis_make_tensor_ptr
etiotto Oct 9, 2024
a7a9b06
Merge branch 'main' into etiotto/axis_analysis_make_tensor_ptr
etiotto Oct 10, 2024
6bddd5f
Merge branch 'main' into etiotto/axis_analysis_make_tensor_ptr
etiotto Oct 10, 2024
4ad4f1a
Merge branch 'main' into etiotto/axis_analysis_make_tensor_ptr
etiotto Oct 10, 2024
4dc1cf1
WIP: Coalescing for block ptrs
etiotto Oct 16, 2024
fa53ced
Fix pre_commit
etiotto Oct 16, 2024
049ddb8
Merge branch 'main' into etiotto/coalesce_for_block_ptr
etiotto Oct 17, 2024
041e2da
Merge branch 'main' into etiotto/coalesce_for_block_ptr
etiotto Oct 17, 2024
5a6cf81
Fix functional problem and add lit test
etiotto Oct 17, 2024
2546665
Fix pre_commit
etiotto Oct 17, 2024
4d5dc49
Reenable rewrite tensor ptr
etiotto Oct 17, 2024
c3fdbba
Fix test_core regression
etiotto Oct 18, 2024
d9de8e7
Fix tutorial assertion
etiotto Oct 18, 2024
949256e
Refactor
etiotto Oct 18, 2024
754ec70
Cleanup
etiotto Oct 18, 2024
469407b
Cleanup
etiotto Oct 18, 2024
9f4f98d
Extend axis info analysis to more block ptrs
etiotto Oct 21, 2024
a40844b
Merge branch 'main' into etiotto/coalesce_for_block_ptr
etiotto Oct 21, 2024
bb9b4c3
Address code review comments
etiotto Oct 22, 2024
8d9a158
Remove unrelated change
etiotto Oct 22, 2024
6529f04
Remove unrelated change
etiotto Oct 22, 2024
0aa334b
Remove unrelated change
etiotto Oct 22, 2024
547d6fa
Fix pre_commit
etiotto Oct 22, 2024
6566f6c
Merge branch 'main' into etiotto/coalesce_for_block_ptr
etiotto Oct 23, 2024
2f97c1a
Address code review comments
etiotto Oct 23, 2024
95f5832
Fix pre_commit
etiotto Oct 23, 2024
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 4 additions & 2 deletions test/Analysis/intel/test-axis-info.mlir
Original file line number Diff line number Diff line change
Expand Up @@ -885,9 +885,11 @@ tt.func public @make_tensor_ptr(%arg0: !tt.ptr<f16>, %arg1: !tt.ptr<f8E5M2> {tt.
%c1_i64 = arith.constant 1 : i64
%c32_i64 = arith.constant 32 : i64
%c128_i64 = arith.constant 128 : i64
// CHECK: %0 = tt.make_tensor_ptr %arg0, {{.*}} => contiguity = [128, 32], divisibility = [1, 1], constancy = [1, 1], constant_value = <none>
// CHECK: tt.make_tensor_ptr %arg0, {{.*}} => contiguity = [128, 32], divisibility = [1, 1], constancy = [1, 1], constant_value = <none>
%0 = tt.make_tensor_ptr %arg0, [%c128_i64, %c32_i64], [%c1_i64, %c1_i64], [%c0_i32, %c0_i32] {order = array<i32: 1, 0>} : !tt.ptr<tensor<128x32xf16>>
// CHECK: %1 = tt.make_tensor_ptr %arg1, {{.*}} => contiguity = [32, 1], divisibility = [16, 1], constancy = [1, 1], constant_value = <none>
// CHECK: tt.make_tensor_ptr %arg1, {{.*}} => contiguity = [64, 1], divisibility = [16, 1], constancy = [1, 1], constant_value = <none>
%1 = tt.make_tensor_ptr %arg1, [%c32_i64, %c32_i64], [%c1_i64, %arg2], [%c0_i32, %c0_i32] {order = array<i32: 0, 1>} : <tensor<64x16xf8E5M2>>
// CHECK: tt.make_tensor_ptr %arg1, {{.*}} => contiguity = [32, 64], divisibility = [1, 1], constancy = [1, 1], constant_value = <none>
%2 = tt.make_tensor_ptr %arg1, [%arg2, %c128_i64], [%c1_i64, %c1_i64], [%c0_i32, %c0_i32] {order = array<i32: 0, 1>} : <tensor<32x64xf8E5M2>>
tt.return
}
Loading