Directory listing for /PyTorch/test/pytorch/third_party/flash-attention/csrc/layer_norm/

• ln.h
• ln_api.cpp
• ln_bwd_1024.cu
• ln_bwd_1280.cu
• ln_bwd_1536.cu
• ln_bwd_2048.cu
• ln_bwd_256.cu
• ln_bwd_2560.cu
• ln_bwd_3072.cu
• ln_bwd_4096.cu
• ln_bwd_512.cu
• ln_bwd_5120.cu
• ln_bwd_6144.cu
• ln_bwd_7168.cu
• ln_bwd_768.cu
• ln_bwd_8192.cu
• ln_bwd_kernels.cuh
• ln_fwd_1024.cu
• ln_fwd_1280.cu
• ln_fwd_1536.cu
• ln_fwd_2048.cu
• ln_fwd_256.cu
• ln_fwd_2560.cu
• ln_fwd_3072.cu
• ln_fwd_4096.cu
• ln_fwd_512.cu
• ln_fwd_5120.cu
• ln_fwd_6144.cu
• ln_fwd_7168.cu
• ln_fwd_768.cu
• ln_fwd_8192.cu
• ln_fwd_kernels.cuh
• ln_kernel_traits.h
• ln_parallel_bwd_1024.cu
• ln_parallel_bwd_1280.cu
• ln_parallel_bwd_1536.cu
• ln_parallel_bwd_2048.cu
• ln_parallel_bwd_256.cu
• ln_parallel_bwd_2560.cu
• ln_parallel_bwd_3072.cu
• ln_parallel_bwd_4096.cu
• ln_parallel_bwd_512.cu
• ln_parallel_bwd_5120.cu
• ln_parallel_bwd_6144.cu
• ln_parallel_bwd_7168.cu
• ln_parallel_bwd_768.cu
• ln_parallel_bwd_8192.cu
• ln_parallel_fwd_1024.cu
• ln_parallel_fwd_1280.cu
• ln_parallel_fwd_1536.cu
• ln_parallel_fwd_2048.cu
• ln_parallel_fwd_256.cu
• ln_parallel_fwd_2560.cu
• ln_parallel_fwd_3072.cu
• ln_parallel_fwd_4096.cu
• ln_parallel_fwd_512.cu
• ln_parallel_fwd_5120.cu
• ln_parallel_fwd_6144.cu
• ln_parallel_fwd_7168.cu
• ln_parallel_fwd_768.cu
• ln_parallel_fwd_8192.cu
• ln_parallel_residual_bwd_kernels.cuh
• ln_parallel_residual_fwd_kernels.cuh
• ln_utils.cuh
• README.md
• setup.py
• static_switch.h