Motif-Technologies
/

activation

Model card Files Files and versions

activation / torch-ext

29.5 kB

8 contributors

History: 15 commits

wyldecat's picture

fix(rms_norm.py): add assertion for input gradients to handle unsupported cases in backward pass

f19f8f4 about 2 months ago

activation
fix(rms_norm.py): add assertion for input gradients to handle unsupported cases in backward pass about 2 months ago
torch_binding.cpp

2.05 kB

refactor(activation): change fused_add_rms_norm and fused_add_rms_norm_backward to out-place operations 2 months ago
torch_binding.h

1.77 kB

refactor(activation): change fused_add_rms_norm and fused_add_rms_norm_backward to out-place operations 2 months ago