Home
last modified time | relevance | path

Searched defs:half_t (Results 1 – 25 of 30) sorted by relevance

12

/aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/kernels/
H A DcutlassF_f16_aligned.cu12 … AttentionKernel<cutlass::half_t, cutlass::arch::Sm50, true, 64, 64, 64, true, true>::kNumThreads, in __launch_bounds__()
31 … AttentionKernel<cutlass::half_t, cutlass::arch::Sm70, true, 64, 64, 64, true, true>::kNumThreads, in __launch_bounds__()
50 … AttentionKernel<cutlass::half_t, cutlass::arch::Sm75, true, 64, 64, 64, true, true>::kNumThreads, in __launch_bounds__()
69 … AttentionKernel<cutlass::half_t, cutlass::arch::Sm80, true, 64, 64, 64, true, true>::kNumThreads, in __launch_bounds__()
88 …AttentionKernel<cutlass::half_t, cutlass::arch::Sm50, true, 32, 128, 128, true, true>::kNumThreads, in __launch_bounds__()
107 …AttentionKernel<cutlass::half_t, cutlass::arch::Sm70, true, 32, 128, 128, true, true>::kNumThreads, in __launch_bounds__()
126 …AttentionKernel<cutlass::half_t, cutlass::arch::Sm75, true, 32, 128, 128, true, true>::kNumThreads, in __launch_bounds__()
145 …AttentionKernel<cutlass::half_t, cutlass::arch::Sm80, true, 64, 128, 128, true, true>::kNumThreads, in __launch_bounds__()
164 …AttentionKernel<cutlass::half_t, cutlass::arch::Sm50, true, 32, 128, 65536, true, true>::kNumThrea… in __launch_bounds__()
183 …AttentionKernel<cutlass::half_t, cutlass::arch::Sm70, true, 32, 128, 65536, true, true>::kNumThrea… in __launch_bounds__()
[all …]
H A DcutlassB_f16_aligned_k128.cu12 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, true, false, false, 128, 64, 128, tr… in __launch_bounds__()
31 …AttentionBackwardKernel<cutlass::arch::Sm80, cutlass::half_t, true, false, true, 128, 128, 128, tr… in __launch_bounds__()
50 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, true, false, false, 128, 64, 128>::k… in __launch_bounds__()
69 …AttentionBackwardKernel<cutlass::arch::Sm75, cutlass::half_t, true, false, false, 128, 64, 128>::k… in __launch_bounds__()
88 …AttentionBackwardKernel<cutlass::arch::Sm80, cutlass::half_t, true, false, true, 128, 128, 128>::k… in __launch_bounds__()
107 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, true, false, false, 64, 64, 128, tru… in __launch_bounds__()
126 …AttentionBackwardKernel<cutlass::arch::Sm80, cutlass::half_t, true, false, false, 64, 64, 128, tru… in __launch_bounds__()
145 …AttentionBackwardKernel<cutlass::arch::Sm50, cutlass::half_t, true, false, false, 64, 64, 128>::kN… in __launch_bounds__()
164 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, true, false, false, 64, 64, 128>::kN… in __launch_bounds__()
183 …AttentionBackwardKernel<cutlass::arch::Sm75, cutlass::half_t, true, false, false, 64, 64, 128>::kN… in __launch_bounds__()
[all …]
H A DcutlassF_f16_notaligned.cu12 … AttentionKernel<cutlass::half_t, cutlass::arch::Sm50, false, 64, 64, 64, true, true>::kNumThreads, in __launch_bounds__()
31 … AttentionKernel<cutlass::half_t, cutlass::arch::Sm70, false, 64, 64, 64, true, true>::kNumThreads, in __launch_bounds__()
50 … AttentionKernel<cutlass::half_t, cutlass::arch::Sm75, false, 64, 64, 64, true, true>::kNumThreads, in __launch_bounds__()
69 …AttentionKernel<cutlass::half_t, cutlass::arch::Sm50, false, 32, 128, 128, true, true>::kNumThread… in __launch_bounds__()
88 …AttentionKernel<cutlass::half_t, cutlass::arch::Sm70, false, 32, 128, 128, true, true>::kNumThread… in __launch_bounds__()
107 …AttentionKernel<cutlass::half_t, cutlass::arch::Sm75, false, 32, 128, 128, true, true>::kNumThread… in __launch_bounds__()
126 …AttentionKernel<cutlass::half_t, cutlass::arch::Sm50, false, 32, 128, 65536, true, true>::kNumThre… in __launch_bounds__()
145 …AttentionKernel<cutlass::half_t, cutlass::arch::Sm70, false, 32, 128, 65536, true, true>::kNumThre… in __launch_bounds__()
164 …AttentionKernel<cutlass::half_t, cutlass::arch::Sm75, false, 32, 128, 65536, true, true>::kNumThre… in __launch_bounds__()
H A DcutlassB_f16_aligned_k128_dropout.cu12 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, true, true, false, 128, 64, 128>::kN… in __launch_bounds__()
31 …AttentionBackwardKernel<cutlass::arch::Sm75, cutlass::half_t, true, true, false, 128, 64, 128>::kN… in __launch_bounds__()
50 …AttentionBackwardKernel<cutlass::arch::Sm80, cutlass::half_t, true, true, true, 128, 128, 128>::kN… in __launch_bounds__()
69 …AttentionBackwardKernel<cutlass::arch::Sm50, cutlass::half_t, true, true, false, 64, 64, 128>::kNu… in __launch_bounds__()
88 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, true, true, false, 64, 64, 128>::kNu… in __launch_bounds__()
107 …AttentionBackwardKernel<cutlass::arch::Sm75, cutlass::half_t, true, true, false, 64, 64, 128>::kNu… in __launch_bounds__()
126 …AttentionBackwardKernel<cutlass::arch::Sm80, cutlass::half_t, true, true, false, 64, 64, 128>::kNu… in __launch_bounds__()
H A DcutlassB_f16_aligned_k65536.cu12 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, true, false, false, 128, 64, 65536>:… in __launch_bounds__()
31 …AttentionBackwardKernel<cutlass::arch::Sm75, cutlass::half_t, true, false, false, 128, 64, 65536>:… in __launch_bounds__()
50 …AttentionBackwardKernel<cutlass::arch::Sm80, cutlass::half_t, true, false, false, 128, 64, 65536>:… in __launch_bounds__()
69 …AttentionBackwardKernel<cutlass::arch::Sm50, cutlass::half_t, true, false, false, 64, 64, 65536>::… in __launch_bounds__()
88 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, true, false, false, 64, 64, 65536>::… in __launch_bounds__()
107 …AttentionBackwardKernel<cutlass::arch::Sm75, cutlass::half_t, true, false, false, 64, 64, 65536>::… in __launch_bounds__()
126 …AttentionBackwardKernel<cutlass::arch::Sm80, cutlass::half_t, true, false, false, 64, 64, 65536>::… in __launch_bounds__()
H A DcutlassB_f16_aligned_k65536_dropout.cu12 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, true, true, false, 128, 64, 65536>::… in __launch_bounds__()
31 …AttentionBackwardKernel<cutlass::arch::Sm75, cutlass::half_t, true, true, false, 128, 64, 65536>::… in __launch_bounds__()
50 …AttentionBackwardKernel<cutlass::arch::Sm80, cutlass::half_t, true, true, false, 128, 64, 65536>::… in __launch_bounds__()
69 …AttentionBackwardKernel<cutlass::arch::Sm50, cutlass::half_t, true, true, false, 64, 64, 65536>::k… in __launch_bounds__()
88 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, true, true, false, 64, 64, 65536>::k… in __launch_bounds__()
107 …AttentionBackwardKernel<cutlass::arch::Sm75, cutlass::half_t, true, true, false, 64, 64, 65536>::k… in __launch_bounds__()
126 …AttentionBackwardKernel<cutlass::arch::Sm80, cutlass::half_t, true, true, false, 64, 64, 65536>::k… in __launch_bounds__()
H A DcutlassB_f16_aligned_k64.cu12 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, true, false, false, 64, 64, 64, true… in __launch_bounds__()
31 …AttentionBackwardKernel<cutlass::arch::Sm80, cutlass::half_t, true, false, true, 64, 64, 64, true>… in __launch_bounds__()
50 …AttentionBackwardKernel<cutlass::arch::Sm50, cutlass::half_t, true, false, false, 64, 64, 64>::kNu… in __launch_bounds__()
69 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, true, false, false, 64, 64, 64>::kNu… in __launch_bounds__()
88 …AttentionBackwardKernel<cutlass::arch::Sm75, cutlass::half_t, true, false, false, 64, 64, 64>::kNu… in __launch_bounds__()
107 …AttentionBackwardKernel<cutlass::arch::Sm80, cutlass::half_t, true, false, true, 64, 64, 64>::kNum… in __launch_bounds__()
H A DcutlassB_f16_aligned_k32.cu12 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, true, false, false, 64, 64, 32, true… in __launch_bounds__()
31 …AttentionBackwardKernel<cutlass::arch::Sm80, cutlass::half_t, true, false, true, 64, 64, 32, true>… in __launch_bounds__()
50 …AttentionBackwardKernel<cutlass::arch::Sm50, cutlass::half_t, true, false, false, 64, 64, 32>::kNu… in __launch_bounds__()
69 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, true, false, false, 64, 64, 32>::kNu… in __launch_bounds__()
88 …AttentionBackwardKernel<cutlass::arch::Sm75, cutlass::half_t, true, false, false, 64, 64, 32>::kNu… in __launch_bounds__()
107 …AttentionBackwardKernel<cutlass::arch::Sm80, cutlass::half_t, true, false, true, 64, 64, 32>::kNum… in __launch_bounds__()
H A DcutlassB_f16_notaligned_k65536.cu12 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, false, false, false, 128, 64, 65536>… in __launch_bounds__()
31 …AttentionBackwardKernel<cutlass::arch::Sm75, cutlass::half_t, false, false, false, 128, 64, 65536>… in __launch_bounds__()
50 …AttentionBackwardKernel<cutlass::arch::Sm50, cutlass::half_t, false, false, false, 64, 64, 65536>:… in __launch_bounds__()
69 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, false, false, false, 64, 64, 65536>:… in __launch_bounds__()
88 …AttentionBackwardKernel<cutlass::arch::Sm75, cutlass::half_t, false, false, false, 64, 64, 65536>:… in __launch_bounds__()
H A DcutlassB_f16_notaligned_k65536_dropout.cu12 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, false, true, false, 128, 64, 65536>:… in __launch_bounds__()
31 …AttentionBackwardKernel<cutlass::arch::Sm75, cutlass::half_t, false, true, false, 128, 64, 65536>:… in __launch_bounds__()
50 …AttentionBackwardKernel<cutlass::arch::Sm50, cutlass::half_t, false, true, false, 64, 64, 65536>::… in __launch_bounds__()
69 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, false, true, false, 64, 64, 65536>::… in __launch_bounds__()
88 …AttentionBackwardKernel<cutlass::arch::Sm75, cutlass::half_t, false, true, false, 64, 64, 65536>::… in __launch_bounds__()
H A DcutlassB_f16_notaligned_k128_dropout.cu12 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, false, true, false, 128, 64, 128>::k… in __launch_bounds__()
31 …AttentionBackwardKernel<cutlass::arch::Sm75, cutlass::half_t, false, true, false, 128, 64, 128>::k… in __launch_bounds__()
50 …AttentionBackwardKernel<cutlass::arch::Sm50, cutlass::half_t, false, true, false, 64, 64, 128>::kN… in __launch_bounds__()
69 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, false, true, false, 64, 64, 128>::kN… in __launch_bounds__()
88 …AttentionBackwardKernel<cutlass::arch::Sm75, cutlass::half_t, false, true, false, 64, 64, 128>::kN… in __launch_bounds__()
H A DcutlassB_f16_notaligned_k128.cu12 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, false, false, false, 128, 64, 128>::… in __launch_bounds__()
31 …AttentionBackwardKernel<cutlass::arch::Sm75, cutlass::half_t, false, false, false, 128, 64, 128>::… in __launch_bounds__()
50 …AttentionBackwardKernel<cutlass::arch::Sm50, cutlass::half_t, false, false, false, 64, 64, 128>::k… in __launch_bounds__()
69 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, false, false, false, 64, 64, 128>::k… in __launch_bounds__()
88 …AttentionBackwardKernel<cutlass::arch::Sm75, cutlass::half_t, false, false, false, 64, 64, 128>::k… in __launch_bounds__()
H A DcutlassB_f16_aligned_k32_dropout.cu12 …AttentionBackwardKernel<cutlass::arch::Sm50, cutlass::half_t, true, true, false, 64, 64, 32>::kNum… in __launch_bounds__()
31 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, true, true, false, 64, 64, 32>::kNum… in __launch_bounds__()
50 …AttentionBackwardKernel<cutlass::arch::Sm75, cutlass::half_t, true, true, false, 64, 64, 32>::kNum… in __launch_bounds__()
69 …AttentionBackwardKernel<cutlass::arch::Sm80, cutlass::half_t, true, true, true, 64, 64, 32>::kNumT… in __launch_bounds__()
H A DcutlassB_f16_aligned_k64_dropout.cu12 …AttentionBackwardKernel<cutlass::arch::Sm50, cutlass::half_t, true, true, false, 64, 64, 64>::kNum… in __launch_bounds__()
31 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, true, true, false, 64, 64, 64>::kNum… in __launch_bounds__()
50 …AttentionBackwardKernel<cutlass::arch::Sm75, cutlass::half_t, true, true, false, 64, 64, 64>::kNum… in __launch_bounds__()
69 …AttentionBackwardKernel<cutlass::arch::Sm80, cutlass::half_t, true, true, true, 64, 64, 64>::kNumT… in __launch_bounds__()
H A DcutlassB_f16_notaligned_k32_dropout.cu12 …AttentionBackwardKernel<cutlass::arch::Sm50, cutlass::half_t, false, true, false, 64, 64, 32>::kNu… in __launch_bounds__()
31 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, false, true, false, 64, 64, 32>::kNu… in __launch_bounds__()
50 …AttentionBackwardKernel<cutlass::arch::Sm75, cutlass::half_t, false, true, false, 64, 64, 32>::kNu… in __launch_bounds__()
H A DcutlassB_f16_notaligned_k32.cu12 …AttentionBackwardKernel<cutlass::arch::Sm50, cutlass::half_t, false, false, false, 64, 64, 32>::kN… in __launch_bounds__()
31 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, false, false, false, 64, 64, 32>::kN… in __launch_bounds__()
50 …AttentionBackwardKernel<cutlass::arch::Sm75, cutlass::half_t, false, false, false, 64, 64, 32>::kN… in __launch_bounds__()
H A DcutlassB_f16_notaligned_k64.cu12 …AttentionBackwardKernel<cutlass::arch::Sm50, cutlass::half_t, false, false, false, 64, 64, 64>::kN… in __launch_bounds__()
31 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, false, false, false, 64, 64, 64>::kN… in __launch_bounds__()
50 …AttentionBackwardKernel<cutlass::arch::Sm75, cutlass::half_t, false, false, false, 64, 64, 64>::kN… in __launch_bounds__()
H A DcutlassB_f16_notaligned_k64_dropout.cu12 …AttentionBackwardKernel<cutlass::arch::Sm50, cutlass::half_t, false, true, false, 64, 64, 64>::kNu… in __launch_bounds__()
31 …AttentionBackwardKernel<cutlass::arch::Sm70, cutlass::half_t, false, true, false, 64, 64, 64>::kNu… in __launch_bounds__()
50 …AttentionBackwardKernel<cutlass::arch::Sm75, cutlass::half_t, false, true, false, 64, 64, 64>::kNu… in __launch_bounds__()
H A DcutlassB_f16_aligned_k96.cu12 …AttentionBackwardKernel<cutlass::arch::Sm80, cutlass::half_t, true, false, true, 128, 64, 96>::kNu… in __launch_bounds__()
/aosp_15_r20/external/sdv/vsomeip/third_party/boost/asio/include/boost/asio/detail/
Dsolaris_fenced_block.hpp35 enum half_t { half }; enum in boost::asio::detail::solaris_fenced_block
39 explicit solaris_fenced_block(half_t) in solaris_fenced_block() argument
Dmacos_fenced_block.hpp35 enum half_t { half }; enum in boost::asio::detail::macos_fenced_block
39 explicit macos_fenced_block(half_t) in macos_fenced_block() argument
Dstd_fenced_block.hpp35 enum half_t { half }; enum in boost::asio::detail::std_fenced_block
39 explicit std_fenced_block(half_t) in std_fenced_block() argument
Dwin_fenced_block.hpp35 enum half_t { half }; enum in boost::asio::detail::win_fenced_block
39 explicit win_fenced_block(half_t) in win_fenced_block() argument
Dgcc_arm_fenced_block.hpp34 enum half_t { half }; enum in boost::asio::detail::gcc_arm_fenced_block
38 explicit gcc_arm_fenced_block(half_t) in gcc_arm_fenced_block() argument
Dgcc_hppa_fenced_block.hpp34 enum half_t { half }; enum in boost::asio::detail::gcc_hppa_fenced_block
38 explicit gcc_hppa_fenced_block(half_t) in gcc_hppa_fenced_block() argument

12