Skip to content

Commit

Permalink
Bump 3rdparty/cutlass_fpA_intB_gemm version (#16506)
Browse files Browse the repository at this point in the history
  • Loading branch information
@vinx13
vinx13 authored Feb 3, 2024
1 parent cdc2303 commit a6e8fee
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion 3rdparty/cutlass_fpA_intB_gemm
Submodule cutlass_fpA_intB_gemm updated 14 files
+8 −4 cutlass_extensions/include/cutlass_extensions/gemm/kernel/fpA_intB_gemm_with_broadcast.h
+2 −2 cutlass_extensions/include/cutlass_extensions/gemm/kernel/mixed_gemm_B_layout.h
+1 −1 cutlass_extensions/include/cutlass_extensions/gemm/kernel/moe_cutlass_kernel.h
+2 −3 cutlass_kernels/fpA_intB_gemm.h
+1 −1 cutlass_kernels/fpA_intB_gemm/fpA_intB_gemm.h
+6 −5 cutlass_kernels/fpA_intB_gemm/fpA_intB_gemm_template.h
+5 −8 cutlass_kernels/moe_gemm/moe_gemm_kernels.h
+9 −4 cutlass_kernels/moe_gemm/moe_gemm_kernels_fp16_fp16.cu
+10 −5 cutlass_kernels/moe_gemm/moe_gemm_kernels_fp16_uint4.cu
+9 −4 cutlass_kernels/moe_gemm/moe_gemm_kernels_fp16_uint8.cu
+50 −0 cutlass_kernels/moe_gemm/moe_gemm_kernels_template.h
+179 −0 cutlass_kernels/moe_gemm/moe_gemv_kernels.cu
+4 −2 cutlass_kernels/moe_gemm/moe_gemv_kernels.h
+73 −7 tvm_binding/tvm_binding.cu

0 comments on commit a6e8fee

Please sign in to comment.