lll143653
d355b6dc5b
Add warning and recommend to use pnnx ( #5588 )
1 year ago
nihui
3ee5c18f84
pnnx logaddexp ( #5598 )
1 year ago
nihui
e82015878c
Update modelwriter.h for mha scale param
1 year ago
nihui
f825d3a23c
pnnx fuse onnx sdpa pattern and ncnn qdim mha fusion ( #5589 )
1 year ago
nihui
569617f212
pnnx convert onnx expand/permute/repeat/reshape/select/slice/cat/ceil/chunk/flatten/floor/maximum/minimum/split/squeeze/stack/transpose/unbind/unsqueeze ( #5583 )
1 year ago
nihui
e7cae68a22
pnnx convert onnx logsoftmax/logsigmoid/mish/selu/sigmoid/silu/softmin/softplus/softshrink/softsign/tanh/tanhshrink ( #5581 )
1 year ago
nihui
1c40615b2d
pnnx convert onnx sdap reduce min/max/mean/sum/prod ( #5579 )
* pnnx convert onnx sdap
* test reduce
1 year ago
nihui
c59885aeac
pnnx convert onnx multiheadattention ( #5575 )
* pnnx convert onnx multiheadattention
* onnx reducemean reducesum
* reducemax reducemin reduceprod
* mask buggy torch
* avoid shadow output
1 year ago
nihui
854678b5f3
pnnx convert onnx prelu gelu elu leakyrelu relu6 celu hardshrink hardsigmoid hardswish clip ( #5572 )
1 year ago
nihui
d264b6353a
pnnx convert onnx rnn lstm gru ( #5553 )
1 year ago
nihui
74d3eb2345
pnnx convert onnx layernorm instancenorm groupnorm ( #5533 )
* pnnx convert onnx layernorm
* fuse early
* skip layernorm affine false test for torch 2.1
* pnnx convert onnx layernorm instancenorm groupnorm
* take num_features from input shape for instancenorm module
* torch < 1.10 can not handle track_running_stats=True
1 year ago
nihui
98de53920e
pnnx convert onnx lrn, fuse if with constant condition, canonicalize and check foldable inside subgraph ( #5532 )
1 year ago
nihui
755f8e1a6b
pnnx convert onnx batchnorm ( #5529 )
1 year ago
nihui
109ed2665c
pnnx convert onnx avgpool maxpool ( #5527 )
* pnnx convert onnx avgpool
* fuse gather indices
* generalized conv convtranspose
* skip maxpool dilation test for torch < 1.12
* ceil mode for opset 9
1 year ago
nihui
f56b18aaf0
pnnx convert onnx resize upsample ( #5522 )
1 year ago
nihui
2828e7ae96
pnnx reset onnx input shape, convert torch.tile torch.where ( #5517 )
* pnnx reset onnx input shape
* eliminate noop cast
1 year ago
nihui
ffb2fe60ee
pnnx convert onnx pad linear sigmoid softmax relu ( #5516 )
* pnnx convert onnx pad
* pnnx convert onnx linear sigmoid softmax relu
* old onnx softmax
1 year ago
nihui
21babb7eed
pnnx convert onnx conv convtranspose ( #5515 )
1 year ago
nihui
75ba2a01c9
pnnx fill tensor data for ort inference, fix output shape2 mismatch ( #5488 )
2 years ago
nihui
5b49d0aa0a
pnnx onnx fold partial shape constant, fix dce node lost bug ( #5486 )
2 years ago
nihui
740f437462
pnnx fuse simple onnx shape slice chain, fix squeeze dim list ( #5485 )
2 years ago
nihui
e009c36155
fix onnx bn silu ( #5483 )
2 years ago
nihui
5e9bf90455
pnnx onnx with inputshape ( #5480 )
2 years ago
nihui
f40d5414d5
pnnx build onnx2pnnx ( #5461 )
2 years ago
nihui
1b7e6358c9
pnnx load dynamo onnx of segmentation models ( #5458 )
2 years ago
nihui
432a8d9aa1
pnnx replace ncnn input long type with int ( #5457 )
2 years ago
nihui
ab088e05b8
pnnx do not fold aliased tensors with inplace op ( #5455 )
2 years ago
nihui
6110d41b19
pnnx bf16 data type ( #5456 )
2 years ago
nihui
08b7d99a75
rnn/lstm/gru dynamic quantization ( #5435 )
2 years ago
nihuini
802a625ae2
fix pnnx windows build
2 years ago
nihui
1e4daff075
fix ncnn tools crash with compound layer creation, write gemm weight data ( #5416 )
* fix ncnn tools crash with compound layer creation
* write gemm weight data
2 years ago
nihui
db035d602d
update ncnnoptimize layers, lightmode=false keeps original weight ( #5414 )
2 years ago
nihui
84256b1494
pnnx enhance functionize ( #5387 )
* pnnx fix some undefined dtype
* fix ncnn convdw1d dynamic weight loading
2 years ago
nihui
02ba6766bd
pnnx fix some undefined dtype ( #5382 )
2 years ago
nihui
a55fe1c9cf
pnnx load dynamo onnx ( #5363 )
* split load torchscript sources from pnnx, drop cxxabi hack, link torchscript with whole-archive
* check compiler cxx11 abi
* eliminate some onnx noop
2 years ago
nihui
25c4278346
pnnx fuse static embedding ( #5380 )
2 years ago
nihui
1463e0b07c
pnnx convert torch.slice_scatter ( #5377 )
2 years ago
nihui
a15d491beb
pnnx test torch.arange ( #5376 )
2 years ago
nihui
a2414c7f63
fix pnnx convert torch.stack negative dim ( #5375 )
2 years ago
nihui
cf293ec35e
fix out of index access on pattern match failure, fix #5347 ( #5350 )
2 years ago
nihui
dcf94ae0cd
split load torchscript sources from pnnx, drop cxxabi hack, link torchscript with whole-archive ( #5348 )
2 years ago
nihui
f03c8f9307
build universal pnnx wheel and pnnx entrypoint script ( #5331 )
2 years ago
Molly Sophia
545a3671ee
python pnnx add option to change fp16 parameter ( #5320 )
Signed-off-by: Molly Sophia <mollysophia379@gmail.com>
2 years ago
nihui
40958d3ab3
pnnx support dynamic slice indexes ( #5299 )
* pnnx handle two operands add/sub/rsub variant
* fuse dynamic slice indexes, wip
* pnnx sliceindexes
* reset device may change non-dtype input numeric 5 to 6
* print inf as float
* preserve dtype for generation op
* pnnx convert torch.masked_select
* test masked_select
* test negative slice
2 years ago
hugo-syn
f35eb4b3b8
chore: Fix multiple typos ( #5301 )
Signed-off-by: hugo-syn <hugo.vincent@synacktiv.com>
2 years ago
nihui
a705a24f32
pnnx convert some cudnn conv2d variants ( #5289 )
2 years ago
nihui
7ed252c854
pnnx handle index_put with empty indices and scalar values ( #5288 )
2 years ago
nihui
c557fb6704
pnnx model binary over 4g ( #5274 )
2 years ago
nihui
76dcaa4947
do not eliminate noop math if shape changes, improve torch-2.1 mha attn_mask detection ( #5273 )
2 years ago
nihui
6f84952122
pnnx handle more softmin logsoftmax dtype, fuse static full range slices to tensor_split ( #5253 )
* pnnx handle more softmin logsoftmax dtype, fuse static full range slices to tensor_split
* fix convert nn.Conv2d with none bias tensor
* fix embedding input with batch index zero
2 years ago