nihui
171b9d1bba
use spdx license header, copyright Tencent ( #6152 )
10 months ago
nihui
ff11080db6
pnnx convert onnx MaxPool auto_pad same, torch reshape_as ( #6089 )
1 year ago
nihui
abb7de9a66
pnnx transformers update part2 ( #6072 )
* pnnx update batch 1 mask attention, add more clip test
* fuse same qkv input
* skip onnx clip sdpa
* maintain cache
1 year ago
nihui
d9d9cc403d
pnnx torch logical ops test ( #6069 )
1 year ago
nihui
5f056b8fee
pnnx transformers update, part1 ( #6054 )
* local cache
* skip transformer test on torch<2.1
* zip pnnx
* bool8 to bool
* initialize xlnet weights
* onnx einsum
* onnx reducelogsumexp
* skip tensor as scalar attribute
* skip longformer
1 year ago
nihui
1d9c78c526
update pnnx ci ( #6048 )
1 year ago
nihui
b8602a6587
pnnx test_inference with torch.no_grad() ( #6028 )
1 year ago
nihui
65b3530be1
fix pnnx ncnn resolve reshape batch index ( #6021 )
* handle batch index with dynamic reshape
1 year ago
nihui
76ce92073e
update pnnx ci with torch 2.7 ( #6018 )
1 year ago
nihui
8363040cb4
pnnx ncnn gelu fast mode, fix interp 2d resize ( #5999 )
1 year ago
nihui
ef0b0e631c
interp output size expression ( #5994 )
1 year ago
nihui
39c055d7f2
crop axes starts ends expression ( #5976 )
* skip dynamic tensor index
* handle clone oom
1 year ago
nihui
4e34da40f1
pnnx convert onnx reducel1 reducel2, add onnx norm test ( #5988 )
1 year ago
nihui
3b3b7a94a1
pnnx convert onnx globalavgpool, add onnx adaptive avg/max pool test ( #5987 )
1 year ago
nihui
1d95a9cb41
update pnnx ci with torch 2.6 ( #5972 )
1 year ago
nihui
f3651ce5b8
pnnx convert onnx depth2space, add pixelshuffle pixelunshuffle test ( #5967 )
1 year ago
nihui
524001bf54
pnnx convert onnx conv with asymmatric pads ( #5965 )
* pnnx convert onnx conv with asymmatric pads
* pad + conv1d and conv3d
* onnx conv auto_pad
1 year ago
nihui
07267f2618
softmax 4d test and vulkan, softmax unified elempack optimization for x86 arm riscv ( #5931 )
1 year ago
nihui
6396a732ef
reshape shape expression, drop reshape permute, test reshape oom ( #5918 )
1 year ago
nihui
78aca5d30b
pnnx fix onnx clip conversion, add onnx clamp test, always reserve onnx split outputs, convert onnx mod ( #5834 )
1 year ago
nihui
4b0d2dea24
pnnx auto inputshape from traced inputs ( #5825 )
1 year ago
nihui
1172b04355
pnnx pass level2 priority ( #5791 )
* constantlist is constant
* f pad value none
* skip ncnn istft center=False test for old torch
* fix fuse pad conv1d
1 year ago
nihui
0734b657d9
spectrogram and inverse spectrogram ( #5779 )
* only supports hann, hamming and all-one window
* inverse spectrogram does not support length parameter
* spectrogram always returns torch.view_as_real(out) as ncnn does not support complex typed mat yet
* inverse spectrogram always accepts torch.view_as_complex(in) as ncnn does not support complex typed mat yet
1 year ago
nihui
c1f9e959f5
pnnx torch 2.5 ( #5748 )
1 year ago
nihui
80c78a0e40
pnnx fuse t5-layernorm as rmsnorm ( #5675 )
1 year ago
nihui
fdf0df3079
RMSNorm ( #5630 )
1 year ago
nihui
eb6e084c2d
pnnx convert nn.RMSNorm F.rms_norm ( #5628 )
1 year ago
nihui
ecfd88a11b
pnnx2ncnn convert torch.roll with one or two shifts ( #5623 )
1 year ago
nihui
b9debee8fb
pnnx ci for torch 2.4 ( #5618 )
* update onnx proto
1 year ago
nihui
3ee5c18f84
pnnx logaddexp ( #5598 )
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
5e9bf90455
pnnx onnx with inputshape ( #5480 )
2 years ago
nihui
f40d5414d5
pnnx build onnx2pnnx ( #5461 )
2 years ago
nihui
84256b1494
pnnx enhance functionize ( #5387 )
* pnnx fix some undefined dtype
* fix ncnn convdw1d dynamic weight loading
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