WilliamLian
|
9682d08d96
|
refactor primitive hook function
|
5 years ago |
mindspore-ci-bot
|
e03bd975a9
|
!1898 data dump
Merge pull request !1898 from caifubi/data-dump
|
5 years ago |
WilliamLian
|
ea9b5468bb
|
fix bug of hccl kernel info
|
5 years ago |
mindspore-ci-bot
|
6dcfbbefce
|
!3017 Add ps optimizer kernels
Merge pull request !3017 from ZPaC/add-ps-optimizer-kernel
|
5 years ago |
mindspore-ci-bot
|
95048955ef
|
!2925 Add random normal op at GPU
Merge pull request !2925 from peixu_ren/custom_gpu
|
5 years ago |
caifubi
|
c577952c9a
|
Async Data Dump
|
5 years ago |
ZPaC
|
6bb83ad3e1
|
Add ps optimizer kernels.
|
5 years ago |
ZPaC
|
bb6373d690
|
Add ps embedding lookup kernels
|
5 years ago |
peixu_ren
|
cccb230f7b
|
Add random normal cuda implementation on GPU
|
5 years ago |
mindspore-ci-bot
|
f201bd655c
|
!2979 Add push pull kernel
Merge pull request !2979 from ZPaC/add-push-pull-kernel
|
5 years ago |
ZPaC
|
e0da486e2f
|
Add push pull kernels
|
5 years ago |
mindspore-ci-bot
|
ee199007ed
|
!2942 change reduce node's reduce axis attr whem it's using in special foramt
Merge pull request !2942 from lianliguang/convert-reduce-axis-when-selected-hw-special-format-for-reduce-kernel
|
5 years ago |
mindspore-ci-bot
|
cf5a27e97d
|
!2962 gpu support SmoothL1Loss kernel
Merge pull request !2962 from chenweifeng/smoothl1loss
|
5 years ago |
mindspore-ci-bot
|
d1e7b9770f
|
!2964 Increase the max tensor size
Merge pull request !2964 from jiangzhenguang/Increase-the-max-tensor-size
|
5 years ago |
jzg
|
fb90ff164b
|
increase the max size of tensor.
|
5 years ago |
mindspore-ci-bot
|
7fd22f8ede
|
!2971 Reg op info from local config file
Merge pull request !2971 from jjfeing/master
|
5 years ago |
wilfChen
|
0fdc304a8e
|
gpu support smoothl1loss
|
5 years ago |
jzg
|
94d0d45ab4
|
increase the max size of tensor.
|
5 years ago |
WilliamLian
|
4200925383
|
convert the reduce axis attr when the reduce node selected the special format
|
5 years ago |
jjfeing
|
53277f8c02
|
reg op info from local config file
|
5 years ago |
yujianfeng
|
3fdc3629af
|
Check attr exists before getting it in embeddinglookup cpu kernel
|
5 years ago |
mindspore-ci-bot
|
0c7a5f76a3
|
!2779 gpu support CTCLoss kernel
Merge pull request !2779 from chenweifeng/ctcloss
|
5 years ago |
wilfChen
|
d54154a1f9
|
Gpu support ctcloss kernel
|
5 years ago |
kswang
|
4094c4688d
|
use multi thread for reduce sparse gradient
|
5 years ago |
mindspore-ci-bot
|
b82926138d
|
!2905 two level reduce sparse gradient
Merge pull request !2905 from kisnwang/parallel-reduce-sparse-gradient
|
5 years ago |
kswang
|
a719a9fe3a
|
add two level reduce sparse gradient
|
5 years ago |
liuxiao93
|
2b0ecfd2b1
|
Add TBE op UnsortedSegmentProd for VM.
|
5 years ago |
mindspore-ci-bot
|
f92c4a5312
|
!2862 reselect the kernel after rectify the build info of domask
Merge pull request !2862 from lianliguang/reselect-the-kernel-info-after-rectify-domask-kernel
|
5 years ago |
WilliamLian
|
60e3849178
|
reselect the domask's child node after rectify the node domask
|
5 years ago |
mindspore-ci-bot
|
99d6430b3f
|
!2835 fix cpu gatherV2 index out of bounds
Merge pull request !2835 from baihuawei/gatherV2
|
5 years ago |
baihuawei
|
b1d6ef0e88
|
fix GatherV2 index out of bounds
|
5 years ago |
mindspore-ci-bot
|
ac3c35c329
|
!2749 GPU update argmaxwithvalue
Merge pull request !2749 from VectorSL/argmaxwithvalue
|
5 years ago |
mindspore-ci-bot
|
b052ecf43b
|
!2804 Add TBE operators SparseApplyFtrlV2\SparseApplyAdagradV2 for VM.
Merge pull request !2804 from liuxiao93/ops-for-VM
|
5 years ago |
duxiutao
|
1e43c609e0
|
Add test case and fix two bugs
1. add case to guard precision
2. fix a shape bug
3. fix a funcGraph bug
|
5 years ago |
liuxiao
|
ddedc41613
|
Add TBE operators SparseApplyFtrlV2\SparseApplyAdagradV2 for VM.
|
5 years ago |
mindspore-ci-bot
|
db51772a92
|
!2750 Clean review bot warning.
Merge pull request !2750 from TronZhang/clean_bot
|
5 years ago |
tronzhang
|
cf8ea2cb66
|
clean review bot warning
|
5 years ago |
yujianfeng
|
2a31e52bf1
|
Adjust the thread num to compute optimizer in cpu
|
5 years ago |
VectorSL
|
f49b8ad6d2
|
update argmaxwithvalue
|
5 years ago |
mindspore-ci-bot
|
746ecc2e4a
|
!2701 add sparse no return optimizer ops
Merge pull request !2701 from wangnan39/distinguish_cpu_sparse_operators_from_tbe
|
5 years ago |
wangnan39@huawei.com
|
68bd5cf6a1
|
add cpu sparse optimizer ops with no return
|
5 years ago |
mindspore-ci-bot
|
2291213b8b
|
!2574 Clean codex
Merge pull request !2574 from TronZhang/codex_0624
|
5 years ago |
mindspore-ci-bot
|
1066debc2b
|
!2698 GPU dropout rewrite
Merge pull request !2698 from VectorSL/drop
|
5 years ago |
mindspore-ci-bot
|
51c4f4a499
|
!2683 Add TBE operaters ApplyAddSign\ApplyPowerSign\ApplyGradientDescent\ApplyProximalGradientDescent for VM.
Merge pull request !2683 from liuxiao/ops-for-VM
|
5 years ago |
VectorSL
|
136b456967
|
gpu dropout rewrite
|
5 years ago |
mindspore-ci-bot
|
fafdfa9a03
|
!2696 [bugfix]:Attr of axis of ReduceSum is invalid
Merge pull request !2696 from zyli2020/bug_fix
|
5 years ago |
lizhenyu
|
7bbe183e6f
|
bugfix:Attr of axis of ReduceSum is invalid
|
5 years ago |
mindspore-ci-bot
|
1a78794947
|
!2680 Add ut for cpu kernel implement of sparse optimizer
Merge pull request !2680 from YuJianfeng/ut
|
5 years ago |
liuxiao
|
2097a0e90a
|
Add optimizer operators for VM.
|
5 years ago |
mindspore-ci-bot
|
f6c80b22f9
|
!2658 clear warning
Merge pull request !2658 from baihuawei/reduce
|
5 years ago |