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

[mlir] Apply is_mlir_conv predicate in standalone MLIr offloading #2249

Merged
merged 3 commits into from
Sep 27, 2023

Conversation

krzysz00
Copy link
Contributor

Currently, the is_mlir_conv predicate wasn't being used when offloading standalone convolutions to MLIR on Navi3x, which caused failures relating to being unable to construct the MLIR program when a 3D convlolution was passed in.

Fixes
https://github.com/ROCmSoftwarePlatform/rocMLIR-internal/issues/1153

This commit amends the standalone lowering to use said predicate, as well as to include quant_convolution and quant_dot into the set of operations that get a standalone lowering.

Currently, the is_mlir_conv predicate wasn't being used when
offloading standalone convolutions to MLIR on Navi3x, which caused
failures relating to being unable to construct the MLIR program when a
3D convlolution was passed in.

Fixes
ROCm/rocMLIR-internal#1153

This commit amends the standalone lowering to use said predicate, as
well as to include quant_convolution and quant_dot into the set of
operations that get a standalone lowering.
@codecov
Copy link

codecov bot commented Sep 26, 2023

Codecov Report

Merging #2249 (34a979c) into develop (434a06c) will increase coverage by 0.00%.
Report is 6 commits behind head on develop.
The diff coverage is 100.00%.

❗ Current head 34a979c differs from pull request most recent head b49cf3d. Consider uploading reports for the commit b49cf3d to get more accurate results

@@           Coverage Diff            @@
##           develop    #2249   +/-   ##
========================================
  Coverage    91.53%   91.53%           
========================================
  Files          429      429           
  Lines        16011    16009    -2     
========================================
- Hits         14655    14654    -1     
+ Misses        1356     1355    -1     
Files Coverage Δ
src/rewrite_pooling.cpp 95.00% <100.00%> (+2.40%) ⬆️
src/simplify_reshapes.cpp 98.73% <100.00%> (+0.01%) ⬆️

@causten causten merged commit a761ffa into develop Sep 27, 2023
14 of 15 checks passed
@causten causten deleted the fix-conv3d-on-navi branch September 27, 2023 18:35
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.

4 participants