robtaylor-chipflow
/

fused-rms-norm

Model card Files Files and versions

fused-rms-norm / torch-ext

1.39 kB

Ctrl+K

Ctrl+K

1 contributor

History: 1 commit

robtaylor-chipflow's picture

robtaylor-chipflow

Add Metal fused_add_rms_norm + rms_norm kernels for vLLM

6c81401 about 2 months ago

fused_rms_norm
Add Metal fused_add_rms_norm + rms_norm kernels for vLLM about 2 months ago
torch_binding.cpp

562 Bytes
Add Metal fused_add_rms_norm + rms_norm kernels for vLLM about 2 months ago
torch_binding.h

286 Bytes
Add Metal fused_add_rms_norm + rms_norm kernels for vLLM about 2 months ago