remove FLOP computation for 3rd party lib call (#4005)
authorCody Hao Yu <comaniac0422@gmail.com>
Wed, 25 Sep 2019 20:50:42 +0000 (13:50 -0700)
committerYuwei Hu <huyuwei1995@gmail.com>
Wed, 25 Sep 2019 20:50:42 +0000 (13:50 -0700)
topi/python/topi/cuda/dense.py

index 1db56d1..a5c2014 100644 (file)
@@ -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])