Bruce-Lee-LY

Expert
@Bruce-Lee-LY

LLM Infer, AI Infra, CUDA

cuda_hgemm. Several optimization methods of half-precision general matrix multiplication (HGEMM) using tensor core with WMMA API and MMA PTX instruction.

559

cuda_hook. Hooked CUDA-related dynamic libraries by using automated code generation tools.

173

cuda_hgemv. Several optimization methods of half-precision general matrix vector multiplication (HGEMV) using CUDA core.

75

decoding_attention. Decoding Attention is specially optimized for MHA, MQA, GQA and MLA using CUDA core for the decoding stage of LLM inference.

48

flash_attention_inference. Performance of the C++ interface of flash attention and flash attention v2 in large language model (LLM) inference scenarios.

45

cuda_auto_tune. NCU-driven iterative optimization workflow for CUDA/CUTLASS/Triton/CuTe DSL kernels.

24

cutlass_gemm. Multiple GEMM operators are constructed with cutlass to support LLM inference.

20

matrix_multiply. Several common methods of matrix multiplication are implemented on CPU and Nvidia GPU using C++11 and CUDA.

14

cuda_back2back_hgemm. Use tensor core to calculate back-to-back HGEMM (half-precision general matrix multiplication) with MMA PTX instruction.

13

memory_pool. Simple and efficient memory pool is implemented with C++11.

10

DeepGEMMPerTensor. DeepGEMMPerTensor: clean and efficient FP8 GEMM per tensor kernels without scales

5

thread_pool. Thread pool is implemented to process task queue with C++11.

3

deep_learning. Implemented the training and inference of several common deep learning model algorithms with tensorflow and pytorch.

1
13
Apply