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

PYBIND11_MODULE(TORCH_EXTENSION_NAME, m)
{
    MOE_C_PYBIND;
}