FlamingoPg 301cbc1d75 feat: support libtorch

2025-10-21 09:55:33 +00:00

..

epilogue.hpp

Make various updates and fixes (#198)

2025-09-25 16:19:07 +08:00

runtime_utils.hpp

feat: support libtorch

2025-10-21 09:55:33 +00:00

sm90_bf16_gemm.hpp

feat: support libtorch

2025-10-21 09:55:33 +00:00

sm90_bmk_bnk_mn.hpp

feat: support libtorch

2025-10-21 09:55:33 +00:00

sm90_fp8_gemm_1d1d.hpp

feat: support libtorch

2025-10-21 09:55:33 +00:00

sm90_fp8_gemm_1d2d.hpp

feat: support libtorch

2025-10-21 09:55:33 +00:00

sm100_bf16_gemm.hpp

feat: support libtorch

2025-10-21 09:55:33 +00:00

sm100_bmk_bnk_mn.hpp

feat: support libtorch

2025-10-21 09:55:33 +00:00

sm100_fp8_gemm_1d1d.hpp

feat: support libtorch

2025-10-21 09:55:33 +00:00

sm100_fp8_gemm_1d2d.hpp

feat: support libtorch

2025-10-21 09:55:33 +00:00

smxx_clean_logits.hpp

Add SM90 kernels (#200)

2025-09-29 17:00:23 +08:00

smxx_cublaslt.hpp

Make various updates and fixes (#198)

2025-09-25 16:19:07 +08:00

smxx_fp8_mqa_logits.hpp

Add SM90 kernels (#200)

2025-09-29 17:00:23 +08:00

smxx_fp8_paged_mqa_logits.hpp

Add SM100 kernels (#201)

2025-09-29 17:07:28 +08:00

smxx_layout.hpp

feat: support libtorch

2025-10-21 09:55:33 +00:00