Skip to content

Draft: Port three torch2flydsl MoE kernels to FlyDSL 0.2.2#58

Draft
irvineoy wants to merge 1 commit into
mainfrom
codex/pr45-moe-flydsl-port
Draft

Draft: Port three torch2flydsl MoE kernels to FlyDSL 0.2.2#58
irvineoy wants to merge 1 commit into
mainfrom
codex/pr45-moe-flydsl-port

Conversation

@irvineoy

Copy link
Copy Markdown
Collaborator

Summary

Splits the three heavyweight implemented torch2flydsl MoE kernels out of #45 so that the main task-corpus PR is not blocked while their FlyDSL 0.2.2 port is completed:

  • torch2flydsl/moe_a8w4_kernel
  • torch2flydsl/moe_kernel
  • torch2flydsl/moe_swiglu_kernel

This PR contains only the 12 task files for those three tasks.

Dependencies

Current status

Draft: do not merge yet.

The old v0.5.12 runtime fails first on missing FlyDSL APIs. With the intended FlyDSL 0.2.2 runtime, the kernels proceed further but still expose real porting issues, including wrapper/raw-value misuse around arith.shrui and a subsequent invalid arith.extsi i64 -> i32 lowering. These require task code changes rather than a validation-environment workaround.

Merge gate

For each of the three tasks on MI355X:

  • configured compile command passes;
  • every correctness shape passes;
  • full benchmark completes;
  • build/performance_report.json is valid and contains positive target-kernel latency.

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.

1 participant