File tree 2 files changed +2
-1
lines changed
core/conversion/converters/impl
tests/core/conversion/converters
2 files changed +2
-1
lines changed Original file line number Diff line number Diff line change @@ -21,6 +21,7 @@ auto topk_registrations TRTORCH_UNUSED = RegisterNodeConversionPatterns().patter
21
21
auto k = args[1 ].unwrapToInt ();
22
22
auto dim = args[2 ].unwrapToInt ();
23
23
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" );
24
25
// auto sorted = args[4].unwrapToBool(); # Currently unused
25
26
26
27
auto selfDim = util::toVec (self->getDimensions ());
Original file line number Diff line number Diff line change @@ -102,7 +102,7 @@ test_suite(
102
102
":test_stack" ,
103
103
":test_lstm_cell" ,
104
104
":test_unsqueeze" ,
105
- ":test_squeeze"
105
+ ":test_squeeze" ,
106
106
":test_topk" ,
107
107
]
108
108
)
You can’t perform that action at this time.
0 commit comments