Skip to content

GitLab

"platforms/cuda-old/CMakeLists.txt" did not exist on "8f2f6e3963d9b55f7b2e2dabfc475295a35f9ed8"
Switch branch/tag
Find file Normal viewHistoryPermalink
mha_fwd_asm_pybind.cu 157 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 "torch/mha_v3_fwd.h"

PYBIND11_MODULE(TORCH_EXTENSION_NAME, m)
{
    MHA_FWD_ASM_PYBIND;
}