Skip to content

GitLab

"libraries/lepton/vscode:/vscode.git/clone" did not exist on "2127b8ddde7c69985ec999c2e844b18b13046915"
Switch branch/tag
Find file Normal viewHistoryPermalink
moe_sum_pybind.cu 151 Bytes
Newer Older
Xiaowei.zhang's avatar
Init the main branch for aiter  
Xiaowei.zhang committed
1
2
3
4
5
6
7
8
9 
/* SPDX-License-Identifier: MIT
   */
#include "rocm_ops.hpp"
#include "moe_sum.h"

PYBIND11_MODULE(TORCH_EXTENSION_NAME, m)
{
      MOE_SUM_PYBIND;
}