mha_bwd_pybind.cu

// SPDX-License-Identifier: MIT
 
#include "rocm_ops.hpp"
#include "torch/mha_bwd.h"

PYBIND11_MODULE(TORCH_EXTENSION_NAME, m)
{
    MHA_BWD_PYBIND;
}