Source files (~85 under csrc/):
csrc/elementwise/fused_add_rms_norm_kernel.cu ← RMSNorm kernel (Where you encountered the error)
csrc/elementwise/rope.cu ← RoPE (Rotary Position Embedding)
csrc/elementwise/activation.cu ← SiLU/GELU activation functions
csrc/attention/cascade.cu ← Cascade attention
csrc/attention/merge_attn_states.cu ← Attention state merging
csrc/gemm/marlin/gptq_marlin.cu ← Marlin W4A16 GEMM (Core quantization acceleration)
csrc/gemm/gptq/gptq_kernel.cu ← GPTQ dequantization
csrc/gemm/fp8_gemm_kernel.cu ← FP8 matrix multiplication
csrc/speculative/eagle_utils.cu ← EAGLE speculative decoding (tree build + verification)
csrc/speculative/speculative_sampling.cu ← Speculative sampling kernel
csrc/moe/moe_align_kernel.cu ← MoE expert routing
csrc/mamba/causal_conv1d.cu ← Mamba convolution
csrc/common_extension.cc ← Python ↔ C++ binding entry point
... and so on, approx. 85 files
Each file is compiled by nvcc to generate an .o:
build/CMakeFiles/common_ops_sm90_build.dir/csrc/elementwise/fused_add_rms_norm_kernel.cu.o
build/CMakeFiles/common_ops_sm90_build.dir/csrc/gemm/marlin/gptq_marlin.cu.o
... approx. 85 .o files
All .o files are linked into a single .so:
build/sm90/common_ops.abi3.so (~241 MB)