diff --git a/core/conversion/converters/impl/topk.cpp b/core/conversion/converters/impl/topk.cpp index d052dd9ce5..9ca91a2643 100644 --- a/core/conversion/converters/impl/topk.cpp +++ b/core/conversion/converters/impl/topk.cpp @@ -21,6 +21,7 @@ auto topk_registrations TRTORCH_UNUSED = RegisterNodeConversionPatterns().patter auto k = args[1].unwrapToInt(); auto dim = args[2].unwrapToInt(); auto largest = args[3].unwrapToBool(); + LOG_DEBUG("Note: sorted argument is not used in TensorRT for aten::topk, results will depend on the value of largest"); // auto sorted = args[4].unwrapToBool(); # Currently unused auto selfDim = util::toVec(self->getDimensions()); diff --git a/tests/core/conversion/converters/BUILD b/tests/core/conversion/converters/BUILD index 503b92fe7a..8def2280be 100644 --- a/tests/core/conversion/converters/BUILD +++ b/tests/core/conversion/converters/BUILD @@ -102,7 +102,7 @@ test_suite( ":test_stack", ":test_lstm_cell", ":test_unsqueeze", - ":test_squeeze" + ":test_squeeze", ":test_topk", ] )