Commit 92494a8a authored by danyao12's avatar danyao12
Browse files

separate hd pad/unpad kernels

parent b16fa5f0
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
...@@ -19,6 +19,22 @@ extern unsigned char bwd_fp16_a16[]; ...@@ -19,6 +19,22 @@ extern unsigned char bwd_fp16_a16[];
extern unsigned char bwd_fp16_a32[]; extern unsigned char bwd_fp16_a32[];
extern unsigned char bwd_fp16_causal_a16[]; extern unsigned char bwd_fp16_causal_a16[];
extern unsigned char bwd_fp16_causal_a32[]; extern unsigned char bwd_fp16_causal_a32[];
extern unsigned char bwd_bf16_a16_rtna_pddv[];
extern unsigned char bwd_bf16_a16_rtne_pddv[];
extern unsigned char bwd_bf16_a16_rtz_pddv[];
extern unsigned char bwd_bf16_a32_rtna_pddv[];
extern unsigned char bwd_bf16_a32_rtne_pddv[];
extern unsigned char bwd_bf16_a32_rtz_pddv[];
extern unsigned char bwd_bf16_causal_a16_rtna_pddv[];
extern unsigned char bwd_bf16_causal_a16_rtne_pddv[];
extern unsigned char bwd_bf16_causal_a16_rtz_pddv[];
extern unsigned char bwd_bf16_causal_a32_rtna_pddv[];
extern unsigned char bwd_bf16_causal_a32_rtne_pddv[];
extern unsigned char bwd_bf16_causal_a32_rtz_pddv[];
extern unsigned char bwd_fp16_a16_pddv[];
extern unsigned char bwd_fp16_a32_pddv[];
extern unsigned char bwd_fp16_causal_a16_pddv[];
extern unsigned char bwd_fp16_causal_a32_pddv[];
extern unsigned char bwd_hd64_bf16_a16_rtna[]; extern unsigned char bwd_hd64_bf16_a16_rtna[];
extern unsigned char bwd_hd64_bf16_a16_rtne[]; extern unsigned char bwd_hd64_bf16_a16_rtne[];
extern unsigned char bwd_hd64_bf16_a16_rtz[]; extern unsigned char bwd_hd64_bf16_a16_rtz[];
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment