Home
last modified time | relevance | path

Searched defs:bfloat16_t (Results 1 – 15 of 15) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/kernels/
H A DcutlassB_bf16_aligned_k128.cu12 …AttentionBackwardKernel<cutlass::arch::Sm80, cutlass::bfloat16_t, true, false, true, 128, 128, 128… in __launch_bounds__()
31 …AttentionBackwardKernel<cutlass::arch::Sm80, cutlass::bfloat16_t, true, false, true, 128, 128, 128… in __launch_bounds__()
50 …AttentionBackwardKernel<cutlass::arch::Sm80, cutlass::bfloat16_t, true, false, false, 64, 64, 128,… in __launch_bounds__()
69 …AttentionBackwardKernel<cutlass::arch::Sm80, cutlass::bfloat16_t, true, false, false, 64, 64, 128>… in __launch_bounds__()
H A DcutlassF_bf16_aligned.cu12 …AttentionKernel<cutlass::bfloat16_t, cutlass::arch::Sm80, true, 64, 64, 64, true, true>::kNumThrea… in __launch_bounds__()
31 …AttentionKernel<cutlass::bfloat16_t, cutlass::arch::Sm80, true, 64, 128, 128, true, true>::kNumThr… in __launch_bounds__()
50 …AttentionKernel<cutlass::bfloat16_t, cutlass::arch::Sm80, true, 32, 128, 65536, true, true>::kNumT… in __launch_bounds__()
H A DcutlassB_bf16_aligned_k65536.cu12 …AttentionBackwardKernel<cutlass::arch::Sm80, cutlass::bfloat16_t, true, false, false, 128, 64, 655… in __launch_bounds__()
31 …AttentionBackwardKernel<cutlass::arch::Sm80, cutlass::bfloat16_t, true, false, false, 64, 64, 6553… in __launch_bounds__()
H A DcutlassB_bf16_aligned_k65536_dropout.cu12 …AttentionBackwardKernel<cutlass::arch::Sm80, cutlass::bfloat16_t, true, true, false, 128, 64, 6553… in __launch_bounds__()
31 …AttentionBackwardKernel<cutlass::arch::Sm80, cutlass::bfloat16_t, true, true, false, 64, 64, 65536… in __launch_bounds__()
H A DcutlassB_bf16_aligned_k128_dropout.cu12 …AttentionBackwardKernel<cutlass::arch::Sm80, cutlass::bfloat16_t, true, true, true, 128, 128, 128>… in __launch_bounds__()
31 …AttentionBackwardKernel<cutlass::arch::Sm80, cutlass::bfloat16_t, true, true, false, 64, 64, 128>:… in __launch_bounds__()
H A DcutlassB_bf16_aligned_k32.cu12 …AttentionBackwardKernel<cutlass::arch::Sm80, cutlass::bfloat16_t, true, false, true, 64, 64, 32, t… in __launch_bounds__()
31 …AttentionBackwardKernel<cutlass::arch::Sm80, cutlass::bfloat16_t, true, false, true, 64, 64, 32>::… in __launch_bounds__()
H A DcutlassB_bf16_aligned_k64.cu12 …AttentionBackwardKernel<cutlass::arch::Sm80, cutlass::bfloat16_t, true, false, true, 64, 64, 64, t… in __launch_bounds__()
31 …AttentionBackwardKernel<cutlass::arch::Sm80, cutlass::bfloat16_t, true, false, true, 64, 64, 64>::… in __launch_bounds__()
H A DcutlassB_bf16_aligned_k96.cu12 …AttentionBackwardKernel<cutlass::arch::Sm80, cutlass::bfloat16_t, true, false, true, 128, 64, 96>:… in __launch_bounds__()
H A DcutlassB_bf16_aligned_k64_dropout.cu12 …AttentionBackwardKernel<cutlass::arch::Sm80, cutlass::bfloat16_t, true, true, true, 64, 64, 64>::k… in __launch_bounds__()
H A DcutlassB_bf16_aligned_k32_dropout.cu12 …AttentionBackwardKernel<cutlass::arch::Sm80, cutlass::bfloat16_t, true, true, true, 64, 64, 32>::k… in __launch_bounds__()
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r522817/lib/clang/18/include/
Darm_bf16.h14 typedef __bf16 bfloat16_t; typedef
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r530567/lib/clang/19/include/
Darm_bf16.h14 typedef __bf16 bfloat16_t; typedef
/aosp_15_r20/prebuilts/clang-tools/linux-x86/lib64/clang/19/include/
H A Darm_bf16.h14 typedef __bf16 bfloat16_t; typedef
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r530567b/lib/clang/19/include/
Darm_bf16.h14 typedef __bf16 bfloat16_t; typedef
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r536225/lib/clang/19/include/
Darm_bf16.h14 typedef __bf16 bfloat16_t; typedef