Skip to content

GitLab

"platforms/cuda-old/src/kernels/kCalculateCDLJForces.h" did not exist on "9eb027741e7860fddaa7eeb5b130b6dcdbc229b7"
Switch branch/tag
Find file Normal viewHistoryPermalink
rmsnorm_pybind.cu 144 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 "rmsnorm.h"

PYBIND11_MODULE(TORCH_EXTENSION_NAME, m)
{
    RMSNORM_PYBIND;
}