Hugging Face's logo

medmekk
/
rmsnorm_thrust

Model card Files Community
medmekk HF Staff commited on
Commit
a63f4de
·
verified ·
1 Parent(s): 753790d

Update rmsnorm_kernel/rmsnorm.cu

Browse files
Files changed (1) hide show
  1. rmsnorm_kernel/rmsnorm.cu +1 -1
rmsnorm_kernel/rmsnorm.cu CHANGED
@@ -1,4 +1,4 @@
1
- #include <torch/extension.h>
2
  #include <thrust/execution_policy.h>
3
  #include <thrust/for_each.h>
4
  #include <thrust/iterator/counting_iterator.h>
 
1
+ #include <torch/all.h>
2
  #include <thrust/execution_policy.h>
3
  #include <thrust/for_each.h>
4
  #include <thrust/iterator/counting_iterator.h>