moe_utils_pybind.cu 155 Bytes
Newer Older
Xiaowei.zhang's avatar
Xiaowei.zhang committed
1
2
3
4
5
6
7
8
9
/* SPDX-License-Identifier: MIT
   */
#include "rocm_ops.hpp"
#include "moe_utils.h"

PYBIND11_MODULE(TORCH_EXTENSION_NAME, m)
{
      MOE_UTILS_PYBIND;
}