Skip to content

Conversation

@ZhangLirong-amd
Copy link
Contributor

@ZhangLirong-amd ZhangLirong-amd commented Oct 11, 2025

Motivation

Make bool guard return func itself to avoid torch.compile creating wrong fx graph

Technical Details

Test Plan

Test Result

Submission Checklist

@gyohuangxin gyohuangxin requested a review from valarLip October 11, 2025 15:06
@gyohuangxin gyohuangxin changed the title Fix meta mha 950 error to make guard return bool func itself [MI35X] Fix meta mha 950 error to make guard return bool func itself Oct 11, 2025
@valarLip valarLip merged commit 6af5453 into main Oct 12, 2025
19 of 24 checks passed
@valarLip valarLip deleted the zlr/meta_mha_fix branch October 12, 2025 08:23
yzhou103 pushed a commit that referenced this pull request Oct 13, 2025
…1170)

* Make bool guard return func itself

* make all return func

* write querysol fake
valarLip added a commit that referenced this pull request Oct 20, 2025
* add asm bf16_gemm to gemm tuner

* mdf_fileName (#1160)

* Tuning Utility update (#1102)

* update

* fix

* fix2

* update

* update gemm/fmoe tune

* modify README.md

* update gen_instances.py

* fix lint error

* fix typo

* rm env AITER_CONFIG_** in ck gen_instances.py

* update a8w8_blockscale_bpreshuffle

* support to read config from multiple tuned files

* fix error in batch_gemm

---------

Co-authored-by: Ying.Zhou2 <Ying.Zhou2@amd.com>
Co-authored-by: Xin Huang <Xin.Huang@amd.com>

* [MI35X] Fix meta mha 950 error to make guard return bool func itself (#1170)

* Make bool guard return func itself

* make all return func

* write querysol fake

* Migrate tests to mi325-1gpu runner due to the slow network issues on TW cluster (#1166)

* Temporarily using mi355 runners util issues on mi325 runners are fixed (#1176)

* Update sglang_downstream.yaml to use aiter-mi325-1gpu runner (#1181)

* CI: Add Aiter Release Package CI (#1171)

* update gemmTuner

* update tuned_gemm.py

* fix lint error

* update asm maxSplitK and tuned_gemm.csv

* Add AITER_CONFIG_GEMM_BF16 and profile_file params to save all temp results

* fix error

* add padded_m logic and update gemm_common

* refactor get tuned config in tuned_gemm
rm  m < 256 limitation in gemm_op_a4w4.py

* fix lint err

* wrapper the get_gemm_a16w16_config using torch_compile_guard

* fix lint error

* refactor tuned_gemm.py

* wrapper gemm_a16w16 with torch_compile_guard

* fix lint error

* add gen_fake func

* update

* updae GemmTuner --profile_file

---------

Co-authored-by: amd-ruitang3 <145657428+amd-ruitang3@users.noreply.github.com>
Co-authored-by: Lingpeng Jin <103567126+valarLip@users.noreply.github.com>
Co-authored-by: Xin Huang <Xin.Huang@amd.com>
Co-authored-by: ZhangLirong <lirzhang@amd.com>
Co-authored-by: valarLip <340077269@qq.com>
eliotwang pushed a commit to eliotwang/aiter that referenced this pull request Oct 21, 2025
…OCm#1170)

* Make bool guard return func itself

* make all return func

* write querysol fake
eliotwang pushed a commit to eliotwang/aiter that referenced this pull request Oct 21, 2025
* add asm bf16_gemm to gemm tuner

* mdf_fileName (ROCm#1160)

* Tuning Utility update (ROCm#1102)

* update

* fix

* fix2

* update

* update gemm/fmoe tune

* modify README.md

* update gen_instances.py

* fix lint error

* fix typo

* rm env AITER_CONFIG_** in ck gen_instances.py

* update a8w8_blockscale_bpreshuffle

* support to read config from multiple tuned files

* fix error in batch_gemm

---------

Co-authored-by: Ying.Zhou2 <Ying.Zhou2@amd.com>
Co-authored-by: Xin Huang <Xin.Huang@amd.com>

* [MI35X] Fix meta mha 950 error to make guard return bool func itself (ROCm#1170)

* Make bool guard return func itself

* make all return func

* write querysol fake

* Migrate tests to mi325-1gpu runner due to the slow network issues on TW cluster (ROCm#1166)

* Temporarily using mi355 runners util issues on mi325 runners are fixed (ROCm#1176)

* Update sglang_downstream.yaml to use aiter-mi325-1gpu runner (ROCm#1181)

* CI: Add Aiter Release Package CI (ROCm#1171)

* update gemmTuner

* update tuned_gemm.py

* fix lint error

* update asm maxSplitK and tuned_gemm.csv

* Add AITER_CONFIG_GEMM_BF16 and profile_file params to save all temp results

* fix error

* add padded_m logic and update gemm_common

* refactor get tuned config in tuned_gemm
rm  m < 256 limitation in gemm_op_a4w4.py

* fix lint err

* wrapper the get_gemm_a16w16_config using torch_compile_guard

* fix lint error

* refactor tuned_gemm.py

* wrapper gemm_a16w16 with torch_compile_guard

* fix lint error

* add gen_fake func

* update

* updae GemmTuner --profile_file

---------

Co-authored-by: amd-ruitang3 <145657428+amd-ruitang3@users.noreply.github.com>
Co-authored-by: Lingpeng Jin <103567126+valarLip@users.noreply.github.com>
Co-authored-by: Xin Huang <Xin.Huang@amd.com>
Co-authored-by: ZhangLirong <lirzhang@amd.com>
Co-authored-by: valarLip <340077269@qq.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.

3 participants