/aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/kernels/ |
H A D | cutlassF_f16_aligned.cu | 11 __global__ void __launch_bounds__( in __launch_bounds__() function 30 __global__ void __launch_bounds__( in __launch_bounds__() function 49 __global__ void __launch_bounds__( in __launch_bounds__() function 68 __global__ void __launch_bounds__( in __launch_bounds__() function 87 __global__ void __launch_bounds__( in __launch_bounds__() function 106 __global__ void __launch_bounds__( in __launch_bounds__() function 125 __global__ void __launch_bounds__( in __launch_bounds__() function 144 __global__ void __launch_bounds__( in __launch_bounds__() function 163 __global__ void __launch_bounds__( in __launch_bounds__() function 182 __global__ void __launch_bounds__( in __launch_bounds__() function [all …]
|
H A D | cutlassF_f32_aligned.cu | 11 __global__ void __launch_bounds__( in __launch_bounds__() function 30 __global__ void __launch_bounds__( in __launch_bounds__() function 49 __global__ void __launch_bounds__( in __launch_bounds__() function 68 __global__ void __launch_bounds__( in __launch_bounds__() function 87 __global__ void __launch_bounds__( in __launch_bounds__() function 106 __global__ void __launch_bounds__( in __launch_bounds__() function 125 __global__ void __launch_bounds__( in __launch_bounds__() function 144 __global__ void __launch_bounds__( in __launch_bounds__() function 163 __global__ void __launch_bounds__( in __launch_bounds__() function 182 __global__ void __launch_bounds__( in __launch_bounds__() function [all …]
|
H A D | cutlassB_f16_aligned_k128.cu | 11 __global__ void __launch_bounds__( in __launch_bounds__() function 30 __global__ void __launch_bounds__( in __launch_bounds__() function 49 __global__ void __launch_bounds__( in __launch_bounds__() function 68 __global__ void __launch_bounds__( in __launch_bounds__() function 87 __global__ void __launch_bounds__( in __launch_bounds__() function 106 __global__ void __launch_bounds__( in __launch_bounds__() function 125 __global__ void __launch_bounds__( in __launch_bounds__() function 144 __global__ void __launch_bounds__( in __launch_bounds__() function 163 __global__ void __launch_bounds__( in __launch_bounds__() function 182 __global__ void __launch_bounds__( in __launch_bounds__() function [all …]
|
H A D | cutlassF_f16_notaligned.cu | 11 __global__ void __launch_bounds__( in __launch_bounds__() function 30 __global__ void __launch_bounds__( in __launch_bounds__() function 49 __global__ void __launch_bounds__( in __launch_bounds__() function 68 __global__ void __launch_bounds__( in __launch_bounds__() function 87 __global__ void __launch_bounds__( in __launch_bounds__() function 106 __global__ void __launch_bounds__( in __launch_bounds__() function 125 __global__ void __launch_bounds__( in __launch_bounds__() function 144 __global__ void __launch_bounds__( in __launch_bounds__() function 163 __global__ void __launch_bounds__( in __launch_bounds__() function
|
H A D | cutlassF_f32_notaligned.cu | 11 __global__ void __launch_bounds__( in __launch_bounds__() function 30 __global__ void __launch_bounds__( in __launch_bounds__() function 49 __global__ void __launch_bounds__( in __launch_bounds__() function 68 __global__ void __launch_bounds__( in __launch_bounds__() function 87 __global__ void __launch_bounds__( in __launch_bounds__() function 106 __global__ void __launch_bounds__( in __launch_bounds__() function 125 __global__ void __launch_bounds__( in __launch_bounds__() function 144 __global__ void __launch_bounds__( in __launch_bounds__() function 163 __global__ void __launch_bounds__( in __launch_bounds__() function
|
H A D | cutlassB_f16_aligned_k128_dropout.cu | 11 __global__ void __launch_bounds__( in __launch_bounds__() function 30 __global__ void __launch_bounds__( in __launch_bounds__() function 49 __global__ void __launch_bounds__( in __launch_bounds__() function 68 __global__ void __launch_bounds__( in __launch_bounds__() function 87 __global__ void __launch_bounds__( in __launch_bounds__() function 106 __global__ void __launch_bounds__( in __launch_bounds__() function 125 __global__ void __launch_bounds__( in __launch_bounds__() function
|
H A D | cutlassB_f16_aligned_k65536.cu | 11 __global__ void __launch_bounds__( in __launch_bounds__() function 30 __global__ void __launch_bounds__( in __launch_bounds__() function 49 __global__ void __launch_bounds__( in __launch_bounds__() function 68 __global__ void __launch_bounds__( in __launch_bounds__() function 87 __global__ void __launch_bounds__( in __launch_bounds__() function 106 __global__ void __launch_bounds__( in __launch_bounds__() function 125 __global__ void __launch_bounds__( in __launch_bounds__() function
|
H A D | cutlassB_f16_aligned_k65536_dropout.cu | 11 __global__ void __launch_bounds__( in __launch_bounds__() function 30 __global__ void __launch_bounds__( in __launch_bounds__() function 49 __global__ void __launch_bounds__( in __launch_bounds__() function 68 __global__ void __launch_bounds__( in __launch_bounds__() function 87 __global__ void __launch_bounds__( in __launch_bounds__() function 106 __global__ void __launch_bounds__( in __launch_bounds__() function 125 __global__ void __launch_bounds__( in __launch_bounds__() function
|
H A D | cutlassB_f16_aligned_k64.cu | 11 __global__ void __launch_bounds__( in __launch_bounds__() function 30 __global__ void __launch_bounds__( in __launch_bounds__() function 49 __global__ void __launch_bounds__( in __launch_bounds__() function 68 __global__ void __launch_bounds__( in __launch_bounds__() function 87 __global__ void __launch_bounds__( in __launch_bounds__() function 106 __global__ void __launch_bounds__( in __launch_bounds__() function
|
H A D | cutlassB_f16_aligned_k32.cu | 11 __global__ void __launch_bounds__( in __launch_bounds__() function 30 __global__ void __launch_bounds__( in __launch_bounds__() function 49 __global__ void __launch_bounds__( in __launch_bounds__() function 68 __global__ void __launch_bounds__( in __launch_bounds__() function 87 __global__ void __launch_bounds__( in __launch_bounds__() function 106 __global__ void __launch_bounds__( in __launch_bounds__() function
|
H A D | cutlassB_f16_notaligned_k65536.cu | 11 __global__ void __launch_bounds__( in __launch_bounds__() function 30 __global__ void __launch_bounds__( in __launch_bounds__() function 49 __global__ void __launch_bounds__( in __launch_bounds__() function 68 __global__ void __launch_bounds__( in __launch_bounds__() function 87 __global__ void __launch_bounds__( in __launch_bounds__() function
|
H A D | cutlassB_f32_aligned_k65536.cu | 11 __global__ void __launch_bounds__( in __launch_bounds__() function 30 __global__ void __launch_bounds__( in __launch_bounds__() function 49 __global__ void __launch_bounds__( in __launch_bounds__() function 68 __global__ void __launch_bounds__( in __launch_bounds__() function 87 __global__ void __launch_bounds__( in __launch_bounds__() function
|
H A D | cutlassB_f32_aligned_k32_dropout.cu | 12 __global__ void __launch_bounds__( in __launch_bounds__() function 32 __global__ void __launch_bounds__( in __launch_bounds__() function 52 __global__ void __launch_bounds__( in __launch_bounds__() function 71 __global__ void __launch_bounds__( in __launch_bounds__() function 90 __global__ void __launch_bounds__( in __launch_bounds__() function
|
H A D | cutlassB_f16_notaligned_k65536_dropout.cu | 11 __global__ void __launch_bounds__( in __launch_bounds__() function 30 __global__ void __launch_bounds__( in __launch_bounds__() function 49 __global__ void __launch_bounds__( in __launch_bounds__() function 68 __global__ void __launch_bounds__( in __launch_bounds__() function 87 __global__ void __launch_bounds__( in __launch_bounds__() function
|
H A D | cutlassB_f16_notaligned_k128_dropout.cu | 11 __global__ void __launch_bounds__( in __launch_bounds__() function 30 __global__ void __launch_bounds__( in __launch_bounds__() function 49 __global__ void __launch_bounds__( in __launch_bounds__() function 68 __global__ void __launch_bounds__( in __launch_bounds__() function 87 __global__ void __launch_bounds__( in __launch_bounds__() function
|
H A D | cutlassB_f32_aligned_k128_dropout.cu | 11 __global__ void __launch_bounds__( in __launch_bounds__() function 30 __global__ void __launch_bounds__( in __launch_bounds__() function 49 __global__ void __launch_bounds__( in __launch_bounds__() function 68 __global__ void __launch_bounds__( in __launch_bounds__() function 87 __global__ void __launch_bounds__( in __launch_bounds__() function
|
H A D | cutlassB_f32_aligned_k65536_dropout.cu | 11 __global__ void __launch_bounds__( in __launch_bounds__() function 30 __global__ void __launch_bounds__( in __launch_bounds__() function 49 __global__ void __launch_bounds__( in __launch_bounds__() function 68 __global__ void __launch_bounds__( in __launch_bounds__() function 87 __global__ void __launch_bounds__( in __launch_bounds__() function
|
H A D | cutlassB_f32_aligned_k64_dropout.cu | 12 __global__ void __launch_bounds__( in __launch_bounds__() function 32 __global__ void __launch_bounds__( in __launch_bounds__() function 52 __global__ void __launch_bounds__( in __launch_bounds__() function 71 __global__ void __launch_bounds__( in __launch_bounds__() function 90 __global__ void __launch_bounds__( in __launch_bounds__() function
|
H A D | cutlassB_f16_notaligned_k128.cu | 11 __global__ void __launch_bounds__( in __launch_bounds__() function 30 __global__ void __launch_bounds__( in __launch_bounds__() function 49 __global__ void __launch_bounds__( in __launch_bounds__() function 68 __global__ void __launch_bounds__( in __launch_bounds__() function 87 __global__ void __launch_bounds__( in __launch_bounds__() function
|
H A D | cutlassB_f32_aligned_k128.cu | 11 __global__ void __launch_bounds__( in __launch_bounds__() function 30 __global__ void __launch_bounds__( in __launch_bounds__() function 49 __global__ void __launch_bounds__( in __launch_bounds__() function 68 __global__ void __launch_bounds__( in __launch_bounds__() function 87 __global__ void __launch_bounds__( in __launch_bounds__() function
|
H A D | cutlassB_f16_aligned_k32_dropout.cu | 11 __global__ void __launch_bounds__( in __launch_bounds__() function 30 __global__ void __launch_bounds__( in __launch_bounds__() function 49 __global__ void __launch_bounds__( in __launch_bounds__() function 68 __global__ void __launch_bounds__( in __launch_bounds__() function
|
H A D | cutlassB_f16_aligned_k64_dropout.cu | 11 __global__ void __launch_bounds__( in __launch_bounds__() function 30 __global__ void __launch_bounds__( in __launch_bounds__() function 49 __global__ void __launch_bounds__( in __launch_bounds__() function 68 __global__ void __launch_bounds__( in __launch_bounds__() function
|
H A D | cutlassB_f32_aligned_k32.cu | 11 __global__ void __launch_bounds__( in __launch_bounds__() function 30 __global__ void __launch_bounds__( in __launch_bounds__() function 49 __global__ void __launch_bounds__( in __launch_bounds__() function 68 __global__ void __launch_bounds__( in __launch_bounds__() function
|
/aosp_15_r20/external/clang/test/SemaCUDA/ |
H A D | launch_bounds.cu | 18 int TestNoFunction __launch_bounds__(128, 7); // expected-warning {{'launch_bounds' attribute only … variable 31 template <int a, int b> __launch_bounds__(a, b) void TestTemplate2Args(void) {} in __launch_bounds__() function 34 template <int a> __launch_bounds__(a) void TestTemplate1Arg(void) {} in __launch_bounds__() function 38 __launch_bounds__(a) void TestTemplate1ArgClass(void) {} // expected-error {{'a' does not refer to … in __launch_bounds__() function 46 __launch_bounds__(Args) void TestTemplateVariadicArgs(void) {} // expected-error {{expression conta… in __launch_bounds__() function
|
/aosp_15_r20/external/clang/test/CodeGenCUDA/ |
H A D | launch-bounds.cu | 11 __launch_bounds__( MAX_THREADS_PER_BLOCK, MIN_BLOCKS_PER_MP ) in __launch_bounds__() function 24 __launch_bounds__( MAX_THREADS_PER_BLOCK ) in __launch_bounds__() function 34 __launch_bounds__(max_threads_per_block) in __launch_bounds__() function 44 __launch_bounds__(max_threads_per_block, min_blocks_per_mp) in __launch_bounds__() function 84 __global__ void __launch_bounds__(constint, constchar) Kernel8() {} in __launch_bounds__() function
|