From 95a4e0611884eadac4a1bfe20603f6232e01d70e Mon Sep 17 00:00:00 2001 From: "lcy.seso" Date: Mon, 16 Sep 2024 10:25:43 +0000 Subject: [PATCH] open the tunning option for tvm. --- .../benchmarks/blocked_sparse_attention/tvm/bigbird_tvm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/artifacts/FractalTensor/benchmarks/blocked_sparse_attention/tvm/bigbird_tvm.py b/artifacts/FractalTensor/benchmarks/blocked_sparse_attention/tvm/bigbird_tvm.py index 5496d6568..5c7e9136f 100644 --- a/artifacts/FractalTensor/benchmarks/blocked_sparse_attention/tvm/bigbird_tvm.py +++ b/artifacts/FractalTensor/benchmarks/blocked_sparse_attention/tvm/bigbird_tvm.py @@ -245,7 +245,7 @@ def bigbird_wv_SPMM_reduce(N, L, B, H, W, R, G, dtype): verbose=2, ) - # tuner.tune(tune_option) + tuner.tune(tune_option) funcs = []