File size: 429 Bytes
a567fa4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
// Copyright (c) Facebook, Inc. and its affiliates.
#include <torch/extension.h>
#include "SwapAlign2Nat/SwapAlign2Nat.h"
namespace tensormask {
PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
m.def(
"swap_align2nat_forward",
&SwapAlign2Nat_forward,
"SwapAlign2Nat_forward");
m.def(
"swap_align2nat_backward",
&SwapAlign2Nat_backward,
"SwapAlign2Nat_backward");
}
} // namespace tensormask
|