Skip to content

Commit 7e7a0fa

Browse files
committed
fix: Patched qconfig_init and get_unwanted_defaults QBmm
Signed-off-by: Brandon Groth <[email protected]>
1 parent 2cb9aca commit 7e7a0fa

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

fms_mo/utils/qconfig_utils.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -232,7 +232,7 @@ def get_recipe(recipe: str, subdir: str = None) -> Union[list, dict]:
232232
return temp_data
233233

234234

235-
def qconfig_init(recipe: str = None, args: Any = None) -> dict:
235+
def qconfig_init(recipe: str = None, args: Any = None, use_mx: bool = False) -> dict:
236236
"""Three possible ways to create qcfg:
237237
1. create a default qcfg
238238
2. load from a json
@@ -725,6 +725,7 @@ def get_unwanted_defaults() -> dict:
725725
nn.ConvTranspose2d: QConvTranspose2d,
726726
nn.Linear: QLinear,
727727
nn.LSTM: QLSTM,
728+
"matmul_or_bmm": QBmm,
728729
},
729730
),
730731
("checkQerr_frequency", False),

0 commit comments

Comments
 (0)