aiter_enum_pybind.cu 179 Bytes
Newer Older
Xiaowei.zhang's avatar
Xiaowei.zhang committed
1
2
3
4
5
6
7
8
9
10
// SPDX-License-Identifier: MIT
 
#include <pybind11/pybind11.h>
#include "rocm_ops.hpp"
#include "aiter_enum.h"

PYBIND11_MODULE(module_aiter_enum, m)
{
    AITER_ENUM_PYBIND;
}