[CUTLASS] Conv2d activation fusion, part 2: Sigmoid fp16, SiLU and HardSwish - #9795

Merged

masahi merged 10 commits into

apache:mainfrom

masahi:cutlass-conv2d-fuse2

Dec 23, 2021

Commits on Dec 22, 2021

[Torch] do not pad if pad widths are all zero
masahi
committed
e0fb117
silu fusion supported
masahi
committed
f23d38d
adding hardswish support
masahi
committed
225b96e
support fast_math sigmoid op
masahi
committed
f56f9bb
fixed type inference for yolov5 + silu fusion
masahi
committed
cbe61c9
use include_non_call_ops=False in AnnotateTarget
masahi
committed
fb57f87
update cutlass
masahi
committed
de3817d
revert change in build.py
masahi
committed
a5d4dd9
simplify codegen
masahi
committed
d47c100
lint
masahi
committed
18e0736