From 5f19e5a880ea224bbd180a8c163b5d980697a99d Mon Sep 17 00:00:00 2001 From: Cody Hao Yu Date: Wed, 25 Sep 2019 13:50:42 -0700 Subject: [PATCH] remove FLOP computation for 3rd party lib call (#4005) --- topi/python/topi/cuda/dense.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/topi/python/topi/cuda/dense.py b/topi/python/topi/cuda/dense.py index 1db56d1..a5c2014 100644 --- a/topi/python/topi/cuda/dense.py +++ b/topi/python/topi/cuda/dense.py @@ -92,10 +92,6 @@ def schedule_dense(cfg, outs): outs = [outs] if isinstance(outs, tvm.tensor.Tensor) else outs if target.target_name == "cuda" and "cublas" in target.libs: - A, B = outs[0].op.input_tensors - b, i = get_const_tuple(A.shape) - o, _ = get_const_tuple(B.shape) - cfg.add_flop(2 * i * b * o) return generic.schedule_extern(outs) s = tvm.create_schedule([x.op for x in outs]) -- 2.7.4