projects
/
platform
/
upstream
/
tvm.git
/ commitdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
| commitdiff |
tree
raw
|
patch
| inline |
side by side
(parent:
f0079a5
)
remove FLOP computation for 3rd party lib call (#4005)
author
Cody Hao Yu
<comaniac0422@gmail.com>
Wed, 25 Sep 2019 20:50:42 +0000
(13:50 -0700)
committer
Yuwei Hu
<huyuwei1995@gmail.com>
Wed, 25 Sep 2019 20:50:42 +0000
(13:50 -0700)
topi/python/topi/cuda/dense.py
patch
|
blob
|
history
diff --git
a/topi/python/topi/cuda/dense.py
b/topi/python/topi/cuda/dense.py
index
1db56d1
..
a5c2014
100644
(file)
--- 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])