Skip to content

T
tic
Commit 5f19e5a8 by Cody Hao Yu Committed by Yuwei Hu
Options

remove FLOP computation for 3rd party lib call (#4005)

parent f0079a57
Showing with 0 additions and 4 deletions
topi/python/topi/cuda/dense.py
...@@ -92,10 +92,6 @@ def schedule_dense(cfg, outs): ...@@ -92,10 +92,6 @@ def schedule_dense(cfg, outs):
outs = [outs] if isinstance(outs, tvm.tensor.Tensor) else outs outs = [outs] if isinstance(outs, tvm.tensor.Tensor) else outs
if target.target_name == "cuda" and "cublas" in target.libs: 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) return generic.schedule_extern(outs)
s = tvm.create_schedule([x.op for x in outs]) s = tvm.create_schedule([x.op for x in outs])
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment