refactor: use __ldlu to load/store data and refactor code for moe permute kernels (#432)

Author: guocuimi

src/kernels/attention/fast_cast.cuh

@@ -4,10 +4,9 @@
 #include <cuda_bf16.h>
 #include <cuda_fp16.h>
 
+#include <cute/numeric/numeric_types.hpp>
 #include <cute/tensor.hpp>
 
-#include "cute/numeric/numeric_types.hpp"
-
 namespace llm {
 
 namespace detail {

src/kernels/moe/CMakeLists.txt

@@ -7,7 +7,7 @@ cc_library(
   SRCS 
     topk_softmax_kernel.cu
     grouped_topk_sigmoid_kernel.cu
-    permute_kernel.cu
+    permutation_index_kernel.cu
   DEPS
     cutlass
     glog::glog
@@ -21,7 +21,7 @@ cc_test(
   SRCS
     topk_softmax_kernel_test.cu
     grouped_topk_sigmoid_kernel_test.cu
-    permute_kernel_test.cu
+    permutation_kernel_test.cu
   DEPS
     :moe.kernels
     absl::random_random