Skip to content

Commit 6dc108a

Browse files
committed
Use 'STABLE' prefix for test library declaration
1 parent 67b3ddb commit 6dc108a

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

src/libtorchaudio/accessor_tests.cpp

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,5 @@
11
#include <libtorchaudio/accessor.h>
22
#include <cstdint>
3-
#include <torch/torch.h>
4-
#include <torch/csrc/stable/tensor.h>
53
#include <torch/csrc/stable/library.h>
64

75
namespace torchaudio {
@@ -33,7 +31,7 @@ void boxed_test_accessor(StableIValue* stack, uint64_t num_args, uint64_t num_ou
3331
stack[0] = from(result);
3432
}
3533

36-
TORCH_LIBRARY_FRAGMENT(torchaudio, m) {
34+
STABLE_TORCH_LIBRARY_FRAGMENT(torchaudio, m) {
3735
m.def(
3836
"_test_accessor(Tensor log_probs) -> bool");
3937
}

0 commit comments

Comments
 (0)