Skip to content

Rebase gfx1250 WMMA/mcast/mxfp4 support onto main #243

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.

Sign up for GitHub

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

Open
jli-melchior wants to merge 11 commits into
base: main
Choose a base branch
from
Open

Rebase gfx1250 WMMA/mcast/mxfp4 support onto main #243

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
11 commits
Select commit Hold shift + click to select a range
0a0f563
squash: pr/v0.1_gfx12 branch changes (87 commits)
jli-melchior Mar 19, 2026
cc9b7e7
refactor multi-stage pipeline
aoli26 Mar 12, 2026
0e5cdbc
pre-calc epilogue addresses to eliminate all s_set_vgpr_msb
aoli26 Mar 12, 2026
3c95640
use ds_load_tr16_b128 to eliminate B bank conflicts
aoli26 Mar 13, 2026
bb8f2a6
support mcast
aoli26 Mar 14, 2026
6d841f4
add mcast unit tests
aoli26 Mar 14, 2026
9fad596
add gemm mxfp4 support
aoli26 Mar 15, 2026
e2735b3
add mxfp4 scale preshuffle optimization
aoli26 Mar 18, 2026
734e317
fix scale preshuffle k-subtile permute and tile-m offset
aoli26 Mar 19, 2026
00ea83c
fix rebase issues: add missing GFX1250 interface methods and fix laun…
jli-melchior Mar 19, 2026
dfe9202
remove local build scripts, ir_dump, and _mlir symlink from repo
jli-melchior Mar 19, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
20 changes: 20 additions & 0 deletions include/flydsl-c/FlyROCDLDialect.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -30,6 +30,26 @@ MLIR_CAPI_EXPORTED MlirType mlirFlyROCDLMmaAtomCDNA3_MFMATypeGetElemTyA(MlirType
MLIR_CAPI_EXPORTED MlirType mlirFlyROCDLMmaAtomCDNA3_MFMATypeGetElemTyB(MlirType type);
MLIR_CAPI_EXPORTED MlirType mlirFlyROCDLMmaAtomCDNA3_MFMATypeGetElemTyAcc(MlirType type);

//===----------------------------------------------------------------------===//
// MmaAtomGFX1250_WMMAType
//===----------------------------------------------------------------------===//

MLIR_CAPI_EXPORTED bool mlirTypeIsAFlyROCDLMmaAtomGFX1250_WMMAType(MlirType type);
MLIR_CAPI_EXPORTED MlirTypeID mlirFlyROCDLMmaAtomGFX1250_WMMATypeGetTypeID(void);

// Constructor
MLIR_CAPI_EXPORTED MlirType mlirFlyROCDLMmaAtomGFX1250_WMMATypeGet(int32_t m, int32_t n, int32_t k,
MlirType elemTyA, MlirType elemTyB,
MlirType elemTyAcc);

// Accessors
MLIR_CAPI_EXPORTED int32_t mlirFlyROCDLMmaAtomGFX1250_WMMATypeGetM(MlirType type);
MLIR_CAPI_EXPORTED int32_t mlirFlyROCDLMmaAtomGFX1250_WMMATypeGetN(MlirType type);
MLIR_CAPI_EXPORTED int32_t mlirFlyROCDLMmaAtomGFX1250_WMMATypeGetK(MlirType type);
MLIR_CAPI_EXPORTED MlirType mlirFlyROCDLMmaAtomGFX1250_WMMATypeGetElemTyA(MlirType type);
MLIR_CAPI_EXPORTED MlirType mlirFlyROCDLMmaAtomGFX1250_WMMATypeGetElemTyB(MlirType type);
MLIR_CAPI_EXPORTED MlirType mlirFlyROCDLMmaAtomGFX1250_WMMATypeGetElemTyAcc(MlirType type);

//===----------------------------------------------------------------------===//
// CopyOpCDNA3BufferLDSTType
//===----------------------------------------------------------------------===//
Expand Down
1 change: 0 additions & 1 deletion include/flydsl/Dialect/Fly/IR/FlyDialect.td
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,6 @@ def Fly_Dialect : Dialect {

let useDefaultTypePrinterParser = 1;
let useDefaultAttributePrinterParser = 1;
let usePropertiesForAttributes = 1;
}

class Fly_Type<string typeName, string typeMnemonic, list<Trait> traits = []>
Expand Down
1 change: 0 additions & 1 deletion include/flydsl/Dialect/FlyROCDL/IR/Dialect.td
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,6 @@ def FlyROCDL_Dialect : Dialect {
];

let useDefaultTypePrinterParser = 1;
let usePropertiesForAttributes = 1;
}

class FlyxROCL_MmaAtom<string typeName, string typeMnemonic, list<Trait> traits = []>
Expand Down
25 changes: 25 additions & 0 deletions include/flydsl/Dialect/FlyROCDL/IR/MmaAtom.td
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -30,4 +30,29 @@ def FlyROCDL_MmaAtomCDNA3_MFMA : FlyxROCL_MmaAtom<"MmaAtomCDNA3_MFMA", "atom.cdn
// MmaAtom CDNA4
//===----------------------------------------------------------------------===//



//===----------------------------------------------------------------------===//
// MmaAtom GFX1250 — WMMA wave32
//===----------------------------------------------------------------------===//

def FlyROCDL_MmaAtomGFX1250_WMMA : FlyxROCL_MmaAtom<"MmaAtomGFX1250_WMMA", "atom.gfx1250.wmma", []> {
let parameters = (ins
"int32_t":$m,
"int32_t":$n,
"int32_t":$k,
"Type":$elemTyA,
"Type":$elemTyB,
"Type":$elemTyAcc
);
let assemblyFormat = "`<` custom<MNKDimensionList>($m, $n, $k) `,` `(` $elemTyA `,` $elemTyB `)` `->` $elemTyAcc `>`";

let builders = [
TypeBuilderWithInferredContext<(ins "int32_t":$m, "int32_t":$n, "int32_t":$k, "Type":$elemTyA, "Type":$elemTyB, "Type":$elemTyAcc), [{
return $_get(elemTyA.getContext(), m, n, k, elemTyA, elemTyB, elemTyAcc);
}]>
];
let genVerifyDecl = 1;
}

#endif // FLYROCDL_MMAATOM
Loading
Loading