feat(aten::topk): Add a debug message noting that sorted is always true

in TensorRT Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
pytorch · Feb 2, 2021 · 81f1e9d · 81f1e9d
1 parent 5bd7528
commit 81f1e9d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git 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
@@ -102,7 +102,7 @@ test_suite(
     ":test_stack",
     ":test_lstm_cell",
     ":test_unsqueeze",
-    ":test_squeeze"
+    ":test_squeeze",
     ":test_topk",
   ]
 )