Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
gaoqiong
composable_kernel_ROCM
Commits
41459498
"git@developer.sourcefind.cn:modelzoo/resnet50_tensorflow.git" did not exist on "2542954295bc91a9ac37ddefade16839e5941ab4"
Commit
41459498
authored
Feb 13, 2025
by
Jim
Browse files
remove code
parent
a651e77b
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
0 additions
and
15 deletions
+0
-15
example/ck_tile/01_fmha/codegen/ops/fmha_bwd.py
example/ck_tile/01_fmha/codegen/ops/fmha_bwd.py
+0
-15
No files found.
example/ck_tile/01_fmha/codegen/ops/fmha_bwd.py
View file @
41459498
...
@@ -989,21 +989,6 @@ class FmhaBwdV3DQDKDVKernel:
...
@@ -989,21 +989,6 @@ class FmhaBwdV3DQDKDVKernel:
F_bf16_cvt
:
int
F_bf16_cvt
:
int
F_is_hdpad
:
str
F_is_hdpad
:
str
# @property
# def gen_bwd_v3_template(self) -> str:
# hdim_template = 64 if self.F_hdim == 64 else 128
# Ts_qo = 32 if hdim_template == 64 else 16
# padding = "t" if self.F_hdim % 64 == 0 else "f"
# padding_suffix = "_hdp" if padding == "t" else ""
# hdim_name = "_hd64" if hdim_template == 64 else ""
# dtype_name = "_{}".format(self.F_dtype)
# causal_name = "_causal" if self.F_is_causal == "t" else ""
# atomic_name = "_a32" if self.F_is_atomic == "t" else "_a16"
# bf16_cvt_name = "_{}".format(BF16_CVT_MAP[self.F_bf16_cvt])
# return FMHA_BWD_V3_TEMPLATE.format(F_hdim=hdim_template, F_dtype=BWD_DTYPE_MAP[self.F_dtype], F_is_atomic=BOOL_MAP[self.F_is_atomic],
# F_is_causal=BOOL_MAP[self.F_is_causal], F_bf16_cvt=self.F_bf16_cvt, F_hdpad=BOOL_MAP[padding], F_Ts_qo = Ts_qo, F_hdim_name=hdim_name,
# F_dtype_name=dtype_name, F_causal_name=causal_name, F_atomic_name=atomic_name, F_bf16_cvt_name=bf16_cvt_name)
def
v3_api_trait
(
self
)
->
FmhaBwdV3DQDKDVApiTrait
:
def
v3_api_trait
(
self
)
->
FmhaBwdV3DQDKDVApiTrait
:
return
FmhaBwdV3DQDKDVApiTrait
(
hdim
=
str
(
self
.
F_hdim
),
return
FmhaBwdV3DQDKDVApiTrait
(
hdim
=
str
(
self
.
F_hdim
),
dtype
=
self
.
F_dtype
,
dtype
=
self
.
F_dtype
,
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment