This repository was archived by the owner on Nov 17, 2023. It is now read-only.

Updated PartialSortSmallK for LT support - #17462

Merged

apeforest merged 10 commits into

apache:masterfrom

JonTanS:LT_PartialSortSmallK

Apr 29, 2020

Commits on Jan 28, 2020

Updated PartialSortSmallK for LT support
jonatan1626
committed
ca3b477

Commits on Jan 30, 2020

Updated Contributors file
jonatan1626
committed
6f209ff

Commits on Feb 13, 2020

Merge branch 'master' into LT_PartialSortSmallK
JonTanS
authored
c77d21b

Commits on Mar 24, 2020

Merge branch 'master' into LT_PartialSortSmallK_test
jonatan1626
committed
4eb32c0
Created a test for LT topk for small values, which wasn't covered by other tests before. Found this issue when running Bert
jonatan1626
committed
0004efb
Updated test utils to accept optional parameter ctx and use cpu ctx if it is None. Also changed topk_small test to be in test_operator.py which contains all the Large Tensor Tests.
jonatan1626
committed
544540b
Removed test topk small
jonatan1626
committed
c1303bd
Updated use of nd to mx.nd
jonatan1626
committed
28319f7

Commits on Apr 23, 2020

Removed tests as original code covers it, but because the memory issue was related to GPU, the tests don't run nightly
tanjna
committed
90abb21
Added back in another line to match the original
tanjna
committed
6b9ce14