Skip to content

Commit d4ea3cf

Browse files
authored
fix (#903)
1 parent a048bf4 commit d4ea3cf

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

op_tests/cpp/mha/compile.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -82,16 +82,16 @@ def compile_mha_bwd(ck_exclude: bool = False): ...
8282
args = parser.parse_args()
8383

8484
if args.api == "fwd":
85-
compile_mha_fwd()
85+
compile_mha_fwd(False)
8686
elif args.api == "bwd":
87-
compile_mha_bwd()
87+
compile_mha_bwd(False)
8888
elif args.api == "fwd_v3":
8989
compile_mha_fwd(True)
9090
elif args.api == "bwd_v3":
9191
compile_mha_bwd(True)
9292
elif args.api == "":
93-
compile_mha_fwd()
94-
compile_mha_bwd()
93+
compile_mha_fwd(False)
94+
compile_mha_bwd(False)
9595
else:
9696
raise ValueError(
9797
"Invalid input value: only support 'fwd', 'bwd' or default to be ''"

0 commit comments

Comments
 (0)