Skip to content

[GEN] Extend Matrix2DBlockLoadOp OCL usages#1259

Merged
etiotto merged 1 commit into
whitneywhtsang/2dblockreadfrom
whitneywhtsang/2dblockread+
Jun 10, 2024
Merged

[GEN] Extend Matrix2DBlockLoadOp OCL usages#1259
etiotto merged 1 commit into
whitneywhtsang/2dblockreadfrom
whitneywhtsang/2dblockread+

Conversation

@whitneywhtsang
Copy link
Copy Markdown
Contributor

No description provided.

@whitneywhtsang whitneywhtsang self-assigned this Jun 5, 2024
@whitneywhtsang whitneywhtsang force-pushed the whitneywhtsang/2dblockread+ branch from 9141e35 to e0f7887 Compare June 5, 2024 20:51
@whitneywhtsang whitneywhtsang changed the title [GEN] Use OCL builtin for 2D block read available on driver 881.12 [GEN] Extend Matrix2DBlockLoadOp OCL usages Jun 5, 2024
@whitneywhtsang whitneywhtsang force-pushed the whitneywhtsang/2dblockread+ branch 2 times, most recently from f7062cb to 4d115e7 Compare June 6, 2024 04:23
@whitneywhtsang whitneywhtsang force-pushed the whitneywhtsang/2dblockread+ branch from 4d115e7 to 33a0aa0 Compare June 6, 2024 04:42
Comment thread test/Conversion/intel/tritongpu_to_llvm_intel_block_ptr.mlir Outdated
Comment thread test/Conversion/intel/tritongpu_to_llvm_intel_block_ptr.mlir Outdated

// CHECK: llvm.func spir_funccc @_Z40intel_sub_group_2d_block_read_8b_8r32x1cPU3AS1viiiDv2_iPt(!llvm.ptr<1>, i32, i32, i32, vector<2xi32>, !llvm.ptr) attributes {passthrough = ["convergent"]}

llvm.func @triton_gen.2Dblockload(%ptr : !llvm.ptr<1>, %base_width : i32, %base_height : i32, %base_pitch : i32, %x : i32, %y : i32) {
Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

These tests are duplicates (only test different variants) and I think is better to limit the number of tests we have to maintain. Add one only please.

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Limited the number of CHECKs except the first test.

@whitneywhtsang whitneywhtsang force-pushed the whitneywhtsang/2dblockread branch from 4e7de79 to 4429340 Compare June 7, 2024 00:33
@whitneywhtsang whitneywhtsang force-pushed the whitneywhtsang/2dblockread+ branch 2 times, most recently from 8ae137b to b504455 Compare June 7, 2024 03:35
@whitneywhtsang whitneywhtsang force-pushed the whitneywhtsang/2dblockread branch 2 times, most recently from f721dde to a24edde Compare June 7, 2024 04:42
@whitneywhtsang whitneywhtsang force-pushed the whitneywhtsang/2dblockread+ branch from b504455 to aebd7b7 Compare June 7, 2024 04:57
@whitneywhtsang whitneywhtsang marked this pull request as ready for review June 7, 2024 05:01
@whitneywhtsang
Copy link
Copy Markdown
Contributor Author

This change depends on coming agama rolling driver.

@whitneywhtsang whitneywhtsang force-pushed the whitneywhtsang/2dblockread+ branch from aebd7b7 to 8df09ed Compare June 7, 2024 06:11
Comment thread test/Conversion/intel/tritongpu_to_llvm_intel_block_ptr.mlir Outdated
Comment thread test/TritonGEN/tritongen-2Dblockload-to-llvm.mlir Outdated
Comment thread test/TritonGEN/tritongen-invalid.mlir Outdated
Comment thread test/TritonGEN/tritongen-to-llvm.mlir Outdated
Comment thread third_party/intel/lib/Dialect/TritonGEN/IR/TritonGENOps.cpp Outdated
@whitneywhtsang whitneywhtsang force-pushed the whitneywhtsang/2dblockread+ branch from 8df09ed to d47921d Compare June 8, 2024 04:23
@whitneywhtsang whitneywhtsang requested a review from etiotto June 8, 2024 04:25
Signed-off-by: Whitney Tsang <whitney.tsang@intel.com>
@whitneywhtsang whitneywhtsang force-pushed the whitneywhtsang/2dblockread+ branch from d47921d to 607d0f4 Compare June 9, 2024 04:50
@etiotto etiotto merged commit 761892b into whitneywhtsang/2dblockread Jun 10, 2024
@etiotto etiotto deleted the whitneywhtsang/2dblockread+ branch June 10, 2024 13:39
whitneywhtsang added a commit that referenced this pull request Jun 10, 2024
Signed-off-by: Whitney Tsang <whitney.tsang@intel.com>
whitneywhtsang added a commit that referenced this pull request Jun 10, 2024
Signed-off-by: Whitney Tsang <whitney.tsang@intel.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

[GEN] Use OCL builtin for 2D block read available on driver 881.12

3 participants