| .. |
|
gemm
|
ex42: Fused MHA imported from xFormers (#662)
|
2022-10-17 10:49:33 -04:00 |
|
iterators
|
ex42: Fused MHA imported from xFormers (#662)
|
2022-10-17 10:49:33 -04:00 |
|
attention_scaling_coefs_updater.h
|
ex42: Fused MHA imported from xFormers (#662)
|
2022-10-17 10:49:33 -04:00 |
|
CMakeLists.txt
|
ex42: Fused MHA imported from xFormers (#662)
|
2022-10-17 10:49:33 -04:00 |
|
debug_utils.h
|
ex42: Fused MHA imported from xFormers (#662)
|
2022-10-17 10:49:33 -04:00 |
|
epilogue_pipelined.h
|
ex42: Fused MHA imported from xFormers (#662)
|
2022-10-17 10:49:33 -04:00 |
|
epilogue_rescale_output.h
|
ex42: Fused MHA imported from xFormers (#662)
|
2022-10-17 10:49:33 -04:00 |
|
epilogue_thread_apply_logsumexp.h
|
ex42: Fused MHA imported from xFormers (#662)
|
2022-10-17 10:49:33 -04:00 |
|
find_default_mma.h
|
ex42: Fused MHA imported from xFormers (#662)
|
2022-10-17 10:49:33 -04:00 |
|
fused_multihead_attention.cu
|
ex42: Fused MHA imported from xFormers (#662)
|
2022-10-17 10:49:33 -04:00 |
|
gemm_kernel_utils.h
|
ex42: Fused MHA imported from xFormers (#662)
|
2022-10-17 10:49:33 -04:00 |
|
kernel_forward.h
|
add #pragma once for header file in example 42 (#698)
|
2022-11-15 22:50:24 -05:00 |
|
mma_from_smem.h
|
ex42: Fused MHA imported from xFormers (#662)
|
2022-10-17 10:49:33 -04:00 |
