Support tilelang backend for torch.

[Xreki]

PR Category

Feature Enhancement

Description

支持TileLang编译器后端，测试方式：

model_path="/work/GraphNet/samples/torchvision/resnet18"
compiler="tilelang"

python -m graph_net_bench.torch.test_compiler \
    --model-path ${model_path} \
    --compiler ${compiler} \
    --device "cuda" \
    --warmup 5 \
    --trials 20

测试样本：
tilelang_example.tar.gz

输入in_0.shape设置成[1024, 1024]使的日志：
log_tilelang_1024x1024.txt

[paddle-bot]

Thanks for your contribution!