![]() * Wrapping code to call the custom op. * Get the rms example to work. * Get around rustfmt failing in the CI. * Fix the rms computation. |
||
---|---|---|
.. | ||
layernorm_kernels.cu | ||
reduction_utils.cuh |
![]() * Wrapping code to call the custom op. * Get the rms example to work. * Get around rustfmt failing in the CI. * Fix the rms computation. |
||
---|---|---|
.. | ||
layernorm_kernels.cu | ||
reduction_utils.cuh |