Add RMS norm (#979)

2023-07-10 21:31:27 -04:00
parent e066ced33b
commit f679663224
3 changed files with 309 additions and 0 deletions
@@ -30,4 +30,5 @@ cutlass_test_unit_add_executable(
  cutlass_test_unit_util
  tensor_reduce.cu
  cutlass_test_levels.cu
+  rms_norm.cu
  )