Commit Graph

930 Commits

Author SHA1 Message Date
danishfarid 722e88995a UpsampleTrilinear3D added
changed to new base class for GPU

working changes + checks

final updates + lint fixes

code check fixes

lint fix 2

lint fix 3

lint fix 4

lint fix final

fix

minor fix

minor fix 2

removed merge marker

address comments

commit fix

comments addressed
2022-06-05 16:52:36 +00:00
i-robot f058997021
!35071 Add feature ReflectionPad1d & ReflecitonPad2d
Merge pull request !35071 from shaojunsong/feature/reflectionPad
2022-06-04 13:00:45 +00:00
shaojunsong c0b9d74987 added reflectionpad1d and 2d 2022-06-04 17:43:49 +08:00
polyhedral 92b1459d57 flatten support functional and tensor interface, dynamic shape and vmap 2022-06-04 11:49:55 +08:00
i-robot 8ae7250a24
!35213 Add broadcast_to op api
Merge pull request !35213 from 朱家兴/master
2022-06-02 03:50:23 +00:00
liuruotao b5b4c7c9d9 takedown gpu testcases where pass >=200 && fail==0 2022-06-01 20:37:11 +08:00
liqiliang d200f89202 Add dynamic_shape tests. 2022-06-01 17:58:33 +08:00
i-robot 6d4921cb1e
!35228 Add Linspace functional API && enable batch support && float64 support
Merge pull request !35228 from zhujingxuan/LinSpace
2022-06-01 01:28:40 +00:00
i-robot 2eefb34afd
!35180 Add vmap and it's tests.
Merge pull request !35180 from liqiliang/vmap_and_dynamic_and_tests
2022-05-31 08:14:15 +00:00
zhujingxuan b92c902a2f Add Linspace functional API && enable batch support && float64 support 2022-05-31 16:08:03 +08:00
i-robot d87af99e7c
!35152 add 1d inputs support for nn.BiDense
Merge pull request !35152 from 吕昱峰(Nate.River)/master
2022-05-31 02:55:46 +00:00
z30020733 942d49e693 Add broadcast_to op api 2022-05-31 10:37:43 +08:00
liqiliang 3f1c71e04a Add vmap and it's tests. 2022-05-31 09:26:13 +08:00
i-robot 922c8cc2ab
!34975 modify vmap rule for masked-select
Merge pull request !34975 from yangzhenzhang/modify-vmap-rule-for-masked-select
2022-05-30 08:31:13 +00:00
i-robot 256303c6ef
!35004 AdaptiveMaxPool2D op.
Merge pull request !35004 from gaoyong10/dynamic_shape_04
2022-05-30 07:36:24 +00:00
i-robot 51c1918011
!34849 add bernoulli ops include func, Tensor interface, GPU kernel and docs.
Merge pull request !34849 from jiangshuqiang/add_bernoulli_
2022-05-30 06:34:08 +00:00
i-robot 6eef2b9826
!35094 add lp_norm vmap and dynamic shape test case.
Merge pull request !35094 from zhuzhongrui/pub_master2
2022-05-30 02:53:52 +00:00
i-robot 8a8985fcb2
!35005 Solve integer division by zero issue for ScatterNdDiv
Merge pull request !35005 from zhujingxuan/scatter
2022-05-30 02:51:44 +00:00
lvyufeng 83f260484e add 1d inputs support for nn.BiDense 2022-05-30 10:11:38 +08:00
i-robot 071ac5425b
!35121 tensor scatter div vmap
Merge pull request !35121 from ling/merge
2022-05-30 01:25:44 +00:00
jiangshuqiang 0449f91f47 add bernoulli ops include func, Tensor interface, GPU kernel and docs. 2022-05-29 13:38:06 +08:00
i-robot ccc8c2744e
!33814 [feat] [assistant] [ops] [I4ZZV4,I4ZZV5] New GPU operator implementation, include tril,triu
Merge pull request !33814 from 路雄博/Tril_GPU
2022-05-29 03:11:33 +00:00
ling 956acdcc13 tensor scatter div vmap
infer shape and datatype bug

tan support GRAPH_MODE  and PYNATIVE_MODE
2022-05-28 17:29:53 +08:00
z00512249 ac4fa51780 add lp_norm vmap and dynamic shape test case. 2022-05-28 16:33:05 +08:00
gaoyong10 cd044f7446 AdaptiveMaxPool2d op. 2022-05-28 15:34:27 +08:00
i-robot ee4048a7f4
!33905 [assistant] [ops] [I4ZZU0, I4ZZU1] New GPU operator implementation, include UpperBound, LowerBound
Merge pull request !33905 from 王超/bound
2022-05-27 08:04:53 +00:00
i-robot 6ff9bc88dd
!34692 [assistant][ops]New operator implementation, include BesselY0, BesselY1, BesselK0, BesselK1, BesselK0e, BesselK1e,BesselI0, BesselKI1, BesselI0e, BesselI1e,BesselJ0,BesselJ1
Merge pull request !34692 from Wangsong95/BesselGPU
2022-05-27 07:29:37 +00:00
zhujingxuan be2c4b99a6 Add zero check for integer division by zero issue 2022-05-27 15:15:28 +08:00
i-robot 3d5403a5f4
!34924 tan vmap
Merge pull request !34924 from ling/pr
2022-05-27 01:11:01 +00:00
王超 8a268f6219 New GPU operator implementation, include UpperBound, LowerBound 2022-05-26 18:30:59 +08:00
ling 5e7c902e34 tan vmap 2022-05-26 16:28:07 +08:00
yangzhenzhang 4ef8e25944 modify vmap rule for masked select 2022-05-26 16:11:32 +08:00
i-robot dde6a2a24e
!34872 Add nn.BiDense
Merge pull request !34872 from 吕昱峰(Nate.River)/master
2022-05-26 07:35:20 +00:00
i-robot c99fa6663c
!34877 Add GPU implementation of SparseSegmentMean operator.
Merge pull request !34877 from hezhenhao1/add_sparse
2022-05-26 07:07:58 +00:00
yelihua 79316f6caf add GPU impl for Fills and testcase 2022-05-25 18:29:10 +08:00
i-robot 5dfd613839
!34447 Support GPU UniqueConsecutive, and add tensor, functional, primitive
Merge pull request !34447 from huangbingjian/unique_consecutive
2022-05-25 09:19:35 +00:00
hezhenhao1 3a2343d3c8 Add GPU implementation of SparseSegmentMean operator. 2022-05-25 17:14:28 +08:00
i-robot 9c9f1a26bd
!34322 Achieve ClipByNorm op on CPU GPU and Ascend
Merge pull request !34322 from JoyLvliang/add_clip_by_norm_op_on_GPU_and_CPU
2022-05-25 02:34:58 +00:00
i-robot 520427cc3a
!34726 ScatterMax support vmap and add dynamic shape case
Merge pull request !34726 from polyhedral/max
2022-05-25 02:14:45 +00:00
lvyufeng 65120caf79 add BiDense 2022-05-25 07:43:35 +08:00
huangbingjian 5a619651d4 Add unique_consecutive ops. 2022-05-24 18:22:07 +08:00
zhangxuetong dbf1c8f18c fix the error of testcase attr 2022-05-24 15:18:55 +08:00
i-robot 0e04a4aaa4
!34769 add tensor_scatter_mul vmap rule && test cases
Merge pull request !34769 from zhuzhongrui/pub_master4
2022-05-24 06:33:34 +00:00
polyhedral fd411c3144 ScatterMax support vmap and add dynamic shape case 2022-05-24 10:27:49 +08:00
7347157+joylvliang@user.noreply.gitee.com 846ea25978 add_clip_by_norm_op_on_GPU_and_CPU 2022-05-24 09:19:00 +08:00
i-robot 54998fab03
!34573 Support vmap for svd
Merge pull request !34573 from huanghui/svd
2022-05-23 08:37:24 +00:00
z00512249 81978165fb add tensor_scatter_mul vmap rule && test cases 2022-05-23 15:42:15 +08:00
i-robot 93a87fd18f
!34665 softshrink softshrinkgrad
Merge pull request !34665 from ling/sr
2022-05-23 06:52:59 +00:00
i-robot b87f5c32d3
!34700 [kernel][gpu]add apply proximal adagrad
Merge pull request !34700 from 张学同/applyproximaladagrad_gpu
2022-05-23 06:47:23 +00:00
i-robot 5a6604775d
!34593 add functional-tensor-vmap impl for masked select
Merge pull request !34593 from yangzhenzhang/add-functional-tensor-vmap-impl-for-masked-select
2022-05-23 06:11:37 +00:00
i-robot 1b3e06faf1
!34645 masked select grad gpu implement
Merge pull request !34645 from zhaodezan/master
2022-05-23 05:53:59 +00:00
i-robot eb856f0d79
!34695 [kernel]fix review comments
Merge pull request !34695 from 张学同/opt_kernel
2022-05-23 03:34:31 +00:00
yangzhenzhang 3b187616e7 add vmap for masked select 2022-05-23 09:10:35 +08:00
zhangxuetong 7463142a05 add apply_proximal_adagrad_gpu_kernel 2022-05-23 09:08:30 +08:00
Signed-off-by: jakcmanftr c4bbbfa6c7 [assistant][ops]New operator implementation for Bessel
Signed-off-by: jakcmanftr <18407825861@163.com>
Signed-off-by: zhangyukun <38148677+jameszhangyukun@users.noreply.github.com>

[assistant][ops]New operator implementation for Bessel

Signed-off-by: jakcmanftr <18407825861@163.com>
Signed-off-by: zhangyukun <38148677+jameszhangyukun@users.noreply.github.com>
2022-05-22 21:35:38 +08:00
zhaodezan 6d9721fbf9 masked select grad gpu first commit 2022-05-22 16:53:34 +08:00
i-robot 68a9ec284d
!33726 [feat] [assistant] [ops] [I4ZZUR] New GPU operator implementation, include RightShift
Merge pull request !33726 from zhixinaa/RightShift
2022-05-21 09:23:36 +00:00
ling 16c573e1ea softshrink softshrinkgrad 2022-05-20 17:09:07 +08:00
zhixinaa d30d43b70d [feat] [assistant] [ops] [I4ZZUR] New GPU operator implementation, include RightShift 2022-05-20 15:33:37 +08:00
limingqi107 817e60a308 add tensor/functional interface for diag op 2022-05-20 15:05:11 +08:00
zhangxuetong c089794e62 fix review comments 2022-05-20 14:53:26 +08:00
i-robot 698ec02a9d
!34670 adapt vmap for scatter_min operator
Merge pull request !34670 from hujiahui8/scatter_min
2022-05-20 02:58:07 +00:00
i-robot 92f73da146
!34616 Tan gpu inference
Merge pull request !34616 from ling/gpu
2022-05-20 01:34:17 +00:00
i-robot 527539ef7c
!34409 unsorted segment prod gpu inference
Merge pull request !34409 from ling/core
2022-05-20 01:20:54 +00:00
i-robot e4148639c4
!34652 [kernel][gpu]add assert kernel
Merge pull request !34652 from 张学同/assert
2022-05-20 01:17:16 +00:00
hujiahui8 2ffa88726c adapt vmap for scatter_min operator 2022-05-19 23:47:44 +08:00
i-robot 4dab24be13
!34479 add gpu diag op
Merge pull request !34479 from limingqi107/bug_fix3
2022-05-19 12:16:51 +00:00
zhangxuetong 66f56e41eb add assert gpu kernel 2022-05-19 19:18:19 +08:00
ling bbbe133362 Tan gpu inference 2022-05-19 19:02:04 +08:00
ling 437deb9da2 unsorted segment prod gpu inference 2022-05-19 18:59:19 +08:00
liangzhibo 46d4c8dc37 Add test case for vmap wrong in_axis of range 2022-05-19 14:49:15 +08:00
i-robot 7c11e3a820
!34480 gpu support atanh kernels
Merge pull request !34480 from chenweifeng/gpu-atanh-kernel
2022-05-19 06:23:32 +00:00
limingqi107 ef55422c04 add gpu diag op 2022-05-19 11:56:13 +08:00
huanghui 636192a116 Support vmap for Svd 2022-05-19 09:29:13 +08:00
wilfChen 1af2656199 gpu support atanh kernel 2022-05-18 15:46:26 +08:00
yangzhenzhang c7a59aa805 add gpu impl of masked select 2022-05-18 09:14:07 +08:00
mengyuanli 49c5b52b9f 1.scatter elements support update input or not
2.add scatter gpu kernel
2022-05-16 20:45:03 +08:00
i-robot 26a1c55fd8
!34334 Support high dimension input for SliceGrad on CPU and GPU
Merge pull request !34334 from xiaotianci/slice_grad
2022-05-16 11:14:17 +00:00
i-robot 9118cf48ca
!33420 Support GPU Svd, and add tensor, functional, primitive
Merge pull request !33420 from huanghui/svd
2022-05-16 02:11:34 +00:00
i-robot 0614c34a9f
!34386 Add a gpu kernel, Padding.
Merge pull request !34386 from liqiliang/padding-gpu
2022-05-16 02:11:13 +00:00
Xiao Tianci 0006a63066 SliceGrad supports 8D input in CPU and GPU 2022-05-16 10:06:22 +08:00
i-robot 9e7b82c9cb
!34324 [MS][OPS] Ger FastGeLU FastGeLUGrad pr3
Merge pull request !34324 from KXiong/master
2022-05-16 01:14:39 +00:00
i-robot c25f2bb443
!34333 Add a gpu kernel, MatrixBandPart.
Merge pull request !34333 from liqiliang/matrixbandpart-gpu
2022-05-14 10:40:00 +00:00
liqiliang 5daf1dd2d4 Add a gpu kernel, Padding. 2022-05-14 17:46:36 +08:00
huanghui cc1f903ce3 1. add Svd Tensor, functional, primitive.
2. support GPU Svd kernel.
2022-05-14 15:34:19 +08:00
i-robot f080d84646
!34374 Update NonZero primitive to support dynamic shape.
Merge pull request !34374 from hezhenhao1/add_nonzero
2022-05-14 06:29:37 +00:00
hezhenhao1 7d9f53b9fb Update NonZero primitive to support dynamic shape. 2022-05-14 11:38:16 +08:00
liqiliang 4be416f40f Add a gpu kernel, MatrixBandPart. 2022-05-14 10:46:38 +08:00
i-robot 377ff71a78
!34166 Add GPU implementation of MatrixDiagV3 operator.
Merge pull request !34166 from hezhenhao1/add_matrix_diag
2022-05-14 02:30:59 +00:00
liangzhibo 1b41721978 Add select tensor, functional interface and vmap 2022-05-13 17:02:21 +08:00
xiongkun f80d30e415 [MS][ops]adjust ger cpu operator
[MS][ops]adjust ger cpu operator

ajust FastGelu and FastGeluGrad

ajust FastGelu and FastGeluGrad

adjust fast_gelu

adjust fast_gelu

adjust fast_gelu

adjust fast_gelu

adjust fast_gelu

adjust bitwise doc

adjust bitwise doc

adjust bitwise doc

adjust bitwise doc

add fast_gelu initsizelist

add fast_gelu initsizelist

ajust fast_gelu_grad_interface

ajust fast_gelu_grad_interface

add fast_gelu doc

add fast_gelu doc

add fast_gelu doc

add fast_gelu doc

add fast_gelu doc

add fast_gelu doc

add fast_gelu doc

add fast_gelu doc

add fast_gelu doc

add FastGeLU FastGeLUGrad vmap

add FastGeLU FastGeLUGrad vmap

add FastGeLU FastGeLUGrad vmap

add FastGeLU FastGeLUGrad vmap

add FastGeLU FastGeLUGrad vmap

add ger vmap

add ger vmap

add ger vmap

add ger vmap

adjust code

adjust code

adjust code

adjust code

adjust code

adjust code

add ger doc

add ger doc

adjust doc

remove bitwise kernelmod

adjust doc

adjust doc

replace tensor

replace tensor

ajust doc

adjust test

adjust test

adjust doc
2022-05-13 15:43:47 +08:00
yuan d04a96e0bb [feat] [assistant] [ops] [I4ZZV4,I4ZZV5] New GPU operator implementation, include tril,triu 2022-05-13 15:16:59 +08:00
hezhenhao1 d5ee2e260c Add GPU implementation of MatrixDiagV3 operator. 2022-05-13 14:57:03 +08:00
i-robot 0968b28871
!34131 scatter_min: add dynamic_shape case for ascend
Merge pull request !34131 from hujiahui8/scatter_min
2022-05-13 06:35:46 +00:00
yangruoqi713 01a06810bf [OP] add function and tensor interface for hard shrink op 2022-05-13 09:37:42 +08:00
i-robot 7c7aa8430a
!34187 Add range functional and vmap
Merge pull request !34187 from LiangZhibo/range_ops
2022-05-12 17:40:17 +00:00
i-robot 4d9e0618c7
!34216 [kernel]add assign sub
Merge pull request !34216 from 张学同/assign_sub
2022-05-12 09:19:23 +00:00
hujiahui8 71f98273ea scatter_min: add dynamic_shape case for ascend 2022-05-12 17:08:02 +08:00
liangzhibo a30897fcfa Add range functional, tensor and vmap 2022-05-12 15:17:59 +08:00
zhangxuetong 8ac51461f5 add assign sub kernel 2022-05-12 09:27:03 +08:00