From: Cody Hao Yu Date: Wed, 25 Sep 2019 20:50:42 +0000 (-0700) Subject: remove FLOP computation for 3rd party lib call (#4005) X-Git-Tag: upstream/0.7.0~1866 X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=5f19e5a880ea224bbd180a8c163b5d980697a99d;p=platform%2Fupstream%2Ftvm.git remove FLOP computation for 3rd party lib call (#4005) --- 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])