We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent a048bf4 commit d4ea3cfCopy full SHA for d4ea3cf
op_tests/cpp/mha/compile.py
@@ -82,16 +82,16 @@ def compile_mha_bwd(ck_exclude: bool = False): ...
82
args = parser.parse_args()
83
84
if args.api == "fwd":
85
- compile_mha_fwd()
+ compile_mha_fwd(False)
86
elif args.api == "bwd":
87
- compile_mha_bwd()
+ compile_mha_bwd(False)
88
elif args.api == "fwd_v3":
89
compile_mha_fwd(True)
90
elif args.api == "bwd_v3":
91
compile_mha_bwd(True)
92
elif args.api == "":
93
94
95
else:
96
raise ValueError(
97
"Invalid input value: only support 'fwd', 'bwd' or default to be ''"
0 commit comments