072c469c060ea1bf6ae409842158d238c64def3f,python/tvm/relay/op/strategy/cuda.py,,topk_strategy_cuda,#Any#Any#Any#Any#,875

Before Change


        wrap_topi_schedule(topi.cuda.schedule_topk),
        name="topk.cuda",
    )
    if target.kind.name == "cuda" and get_global_func(
        "tvm.contrib.thrust.sort", allow_missing=True
    ):
        strategy.add_implementation(
            wrap_compute_topk(topi.cuda.topk_thrust),
            wrap_topi_schedule(topi.cuda.schedule_topk),

After Change


        wrap_topi_schedule(topi.cuda.schedule_topk),
        name="topk.cuda",
    )
    if can_use_thrust(target, "tvm.contrib.thrust.sort"):
        strategy.add_implementation(
            wrap_compute_topk(topi.cuda.topk_thrust),
            wrap_topi_schedule(topi.cuda.schedule_topk),
Italian Trulli
In pattern: SUPERPATTERN

Frequency: 4

Non-data size: 7

Instances


Project Name: apache/incubator-tvm
Commit Name: 072c469c060ea1bf6ae409842158d238c64def3f
Time: 2021-02-22
Author: masahi129@gmail.com
File Name: python/tvm/relay/op/strategy/cuda.py
Class Name:
Method Name: topk_strategy_cuda


Project Name: apache/incubator-tvm
Commit Name: 072c469c060ea1bf6ae409842158d238c64def3f
Time: 2021-02-22
Author: masahi129@gmail.com
File Name: python/tvm/relay/op/strategy/cuda.py
Class Name:
Method Name: scatter_cuda


Project Name: apache/incubator-tvm
Commit Name: 072c469c060ea1bf6ae409842158d238c64def3f
Time: 2021-02-22
Author: masahi129@gmail.com
File Name: python/tvm/relay/op/strategy/cuda.py
Class Name:
Method Name: argsort_strategy_cuda


Project Name: apache/incubator-tvm
Commit Name: 072c469c060ea1bf6ae409842158d238c64def3f
Time: 2021-02-22
Author: masahi129@gmail.com
File Name: python/tvm/relay/op/strategy/cuda.py
Class Name:
Method Name: sort_strategy_cuda