Skip to content

Check if driver support DPAS and 2D block operations#1565

Merged
whitneywhtsang merged 5 commits into
llvm-targetfrom
whitneywhtsang/dev_prop
Jul 9, 2024
Merged

Check if driver support DPAS and 2D block operations#1565
whitneywhtsang merged 5 commits into
llvm-targetfrom
whitneywhtsang/dev_prop

Conversation

@whitneywhtsang
Copy link
Copy Markdown
Contributor

@whitneywhtsang whitneywhtsang commented Jul 4, 2024

Module attributes triton_gpu.support_sg_2d_block and triton_gpu.support_dpas are added to the module if the driver supports the corresponding OpenCL extension.
This way is preferred instead of checking if a driver is LTS, since the OpenCL extension can also be unsupported for non-LTS driver, and checking driver version is error prone.

@whitneywhtsang whitneywhtsang self-assigned this Jul 4, 2024
@whitneywhtsang whitneywhtsang force-pushed the whitneywhtsang/dev_prop branch from 68fdec0 to 5bf94fa Compare July 5, 2024 01:14
Signed-off-by: Whitney Tsang <whitney.tsang@intel.com>
@whitneywhtsang whitneywhtsang force-pushed the whitneywhtsang/dev_prop branch from 5bf94fa to e66c7c4 Compare July 5, 2024 01:59
@whitneywhtsang whitneywhtsang marked this pull request as ready for review July 5, 2024 02:59
@whitneywhtsang whitneywhtsang requested review from a team, chengjunlu, etiotto and leonling-ll July 5, 2024 03:00
Copy link
Copy Markdown

@victor-eds victor-eds left a comment

Choose a reason for hiding this comment

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

Nice improvement! Should the prefix be triton_intel_gpu instead of triton_gpu? Should we have the attribute name be queried from TritonIntelGPUDialect instead of being constants in the code?

Comment thread test/Conversion/intel/arith_to_llvm.mlir Outdated
Comment thread third_party/intel/triton_xpu.cc Outdated
Comment thread third_party/intel/backend/compiler.py Outdated
Signed-off-by: Whitney Tsang <whitney.tsang@intel.com>
@whitneywhtsang whitneywhtsang force-pushed the whitneywhtsang/dev_prop branch from 87482c8 to 5db5c26 Compare July 5, 2024 15:54
Signed-off-by: Whitney Tsang <whitney.tsang@intel.com>
@whitneywhtsang
Copy link
Copy Markdown
Contributor Author

Nice improvement! Should the prefix be triton_intel_gpu instead of triton_gpu? Should we have the attribute name be queried from TritonIntelGPUDialect instead of being constants in the code?

Done.

@whitneywhtsang whitneywhtsang requested a review from victor-eds July 6, 2024 02:56
Comment thread third_party/intel/backend/compiler.py Outdated
Signed-off-by: Whitney Tsang <whitney.tsang@intel.com>
@whitneywhtsang whitneywhtsang force-pushed the whitneywhtsang/dev_prop branch from f047ef6 to 6cff5f6 Compare July 7, 2024 20:26
Copy link
Copy Markdown
Contributor

@leonling-ll leonling-ll left a comment

Choose a reason for hiding this comment

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

LGTM

Copy link
Copy Markdown

@victor-eds victor-eds left a comment

Choose a reason for hiding this comment

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

LGTM after small change

Comment thread third_party/intel/include/Dialect/TritonIntelGPU/IR/TritonIntelGPUDialect.td Outdated
Comment thread third_party/intel/include/TritonIntelGPUToLLVM/TypeConverter.h Outdated
Comment thread third_party/intel/lib/TritonIntelGPUToLLVM/TritonGPUToLLVM.cpp Outdated
Signed-off-by: Whitney Tsang <whitney.tsang@intel.com>
@whitneywhtsang whitneywhtsang force-pushed the whitneywhtsang/dev_prop branch from c0959c6 to 7aab3de Compare July 8, 2024 20:34
@whitneywhtsang whitneywhtsang merged commit 232c1bf into llvm-target Jul 9, 2024
@whitneywhtsang whitneywhtsang deleted the whitneywhtsang/dev_prop branch July 9, 2024 13:03
etiotto added a commit that referenced this pull request Jul 9, 2024
whitneywhtsang added a commit that referenced this pull request Jul 11, 2024
Reland #1565.

---------

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.

5 participants