Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fix a bug in fastPath condition #482

Merged
merged 1 commit into from
Jan 25, 2024
Merged

Fix a bug in fastPath condition #482

merged 1 commit into from
Jan 25, 2024

Conversation

zhanglx13
Copy link

Comment on lines -331 to -334
auto stride0 = findConstValue(smemObj.strides[0]);
auto stride1 = findConstValue(smemObj.strides[1]);
auto offset0 = findConstValue(smemObj.offsets[0]);
auto offset1 = findConstValue(smemObj.offsets[1]);
Copy link
Author

Choose a reason for hiding this comment

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

To those who are interested in the issue
These are no longer constants due to stream-pipeline pass. Therefore, they cannot be used to check is we are going to the fast path.

const MfmaEncodingAttr &dstEncoding) {
if (srcEncoding.getMaxPhase() > 1)
return false;
auto stride0 = findConstValue(smemObj.strides[0]);

Choose a reason for hiding this comment

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

In this case you can remove findConstValue definition, it is not used in other places

Copy link

@binarman binarman left a comment

Choose a reason for hiding this comment

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

LGTM

@zhanglx13 zhanglx13 merged commit 3c6010d into triton-mlir Jan 25, 2024
2 checks passed
zhanglx13 added a commit to triton-lang/triton that referenced this pull request Mar 4, 2024
This PR updates SharedToDotOperandMFMA.cpp and MFMA.cpp.
- SharedToDotOperandMFMA.cpp is up to date with triton-mlir as of today,
which includes changes until ROCm#482
  - Fixed issue with opaque pointers
- Fixed API for `getMFMAElemsPerInstrForOperands` and
`getMFMARepForOperands`
- MFMA.cpp is synced with triton-mlir@6bb04d, which includes changes
until ROCm#469

Note to @binarman: changes in other files from
ROCm#469 are not included in this PR. We
can bring up the support for mfma 64x4 and 4x64 later.
binarman pushed a commit to binarman/triton that referenced this pull request Apr 2, 2024
This PR updates SharedToDotOperandMFMA.cpp and MFMA.cpp.
- SharedToDotOperandMFMA.cpp is up to date with triton-mlir as of today,
which includes changes until ROCm#482
  - Fixed issue with opaque pointers
- Fixed API for `getMFMAElemsPerInstrForOperands` and
`getMFMARepForOperands`
- MFMA.cpp is synced with triton-mlir@6bb04d, which includes changes
until ROCm#469

Note to @binarman: changes in other files from
ROCm#469 are not included in this PR. We
can bring up the support for mfma 64x4 and 4x64 later.
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.

3 participants