Skip to content

Commit 81f1e9d

Browse files
committedFeb 2, 2021
feat(aten::topk): Add a debug message noting that sorted is always true
in TensorRT Signed-off-by: Naren Dasan <naren@narendasan.com> Signed-off-by: Naren Dasan <narens@nvidia.com>
1 parent 5bd7528 commit 81f1e9d

File tree

2 files changed

+2
-1
lines changed
  • core/conversion/converters/impl
  • tests/core/conversion/converters

2 files changed

+2
-1
lines changed
 

‎core/conversion/converters/impl/topk.cpp

+1
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@ auto topk_registrations TRTORCH_UNUSED = RegisterNodeConversionPatterns().patter
2121
auto k = args[1].unwrapToInt();
2222
auto dim = args[2].unwrapToInt();
2323
auto largest = args[3].unwrapToBool();
24+
LOG_DEBUG("Note: sorted argument is not used in TensorRT for aten::topk, results will depend on the value of largest");
2425
// auto sorted = args[4].unwrapToBool(); # Currently unused
2526

2627
auto selfDim = util::toVec(self->getDimensions());

‎tests/core/conversion/converters/BUILD

+1-1
Original file line numberDiff line numberDiff line change
@@ -102,7 +102,7 @@ test_suite(
102102
":test_stack",
103103
":test_lstm_cell",
104104
":test_unsqueeze",
105-
":test_squeeze"
105+
":test_squeeze",
106106
":test_topk",
107107
]
108108
)

0 commit comments

Comments
 (0)
Please sign in to comment.