add ops

torch_mlu_ops-v1.3.2/csrc/kernels/moe/moe.mluh

@@ -0,0 +1,21 @@
+/*************************************************************************
+ * Copyright (C) [2023-2024] by Cambricon, Inc.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
+ * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+ * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
+ * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
+ * CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
+ * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
+ * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+ *************************************************************************/
+#ifndef CSRC_KERNELS_MOE_MOE_MLUH_
+#define CSRC_KERNELS_MOE_MOE_MLUH_
+
+#include "add_bias_activation.mluh"
+#include "combine_result.mluh"
+#include "expand_input.mluh"
+#include "gen_idx.mluh"
+#include "softmax_topk.mluh"
+
+#endif  // CSRC_KERNELS_MOE_MOE_MLUH_