Commit Graph

30060 Commits (rocm_dev_0217)
 

Author SHA1 Message Date
YUNSHEN XIE 3a8ef10e09
fix modified_retry_method_only_win (#31404)
5 years ago
Zhang Ting dcce54ea76
improve performance of depthwise_conv2d (#31099)
5 years ago
wuhuanzhou 4d6d2db812
Windows system supports Ninja compilation (#31161)
5 years ago
liym27 0fff930667
Fix bug for set_value op when input dtype is not float32 (#31411)
5 years ago
Huihuang Zheng c40b98e068
Fix comment (#31424)
5 years ago
Huihuang Zheng 6bf02a1261
[Dy2stat] Fix Read-Only Attribute as while_loop Output (#31415)
5 years ago
jakpiase 5b4f8aac82
Added LSTM BF16 and fixed GRU BF16 (#31234)
5 years ago
Qi Li 7cdf6ea770
[ROCM] update fluid elementwise op for rocm (part10), test=develop (#31361)
5 years ago
Qi Li 84639b6193
[ROCM] update fluid operators for rocm (part3), test=develop (#31213)
5 years ago
Qi Li 3b9db17199
[ROCM] update fluid operators for rocm (part7), test=develop (#31307)
5 years ago
Qi Li db50fb6766
[ROCM] fix softmax with loss and update python scripts, test=develop (#31373)
5 years ago
Pei Yang 32211fe9c4
TRT conv2d converter support SAME padding (#31379)
5 years ago
Qi Li e312a1ff6e
[ROCM] update fluid operators for rocm (part9), test=develop (#31338)
5 years ago
Qi Li 6626c6a6ad
fix bert cu file compiler error, test=develop (#31389)
5 years ago
wuhuanzhou c1bc223695
compile with VS2017, test=develop (#31388)
5 years ago
Zhou Wei 13e4280f82
[Custom OP]polish doc of custom OP (#31369)
5 years ago
Qi Li 946dbdae8c
[ROCM] update fluid operators for rocm (part6), test=develop (#31301)
5 years ago
wangna11BD 1cbccfa594
Add attrs `deformable_groups` for deformable_conv API (#31335)
5 years ago
Shang Zhizhou 77c44e2f1b
change prelu plugin to tensorRT layer (#30210)
5 years ago
YUNSHEN XIE 353dd0cd98
Modified retry method on windows (#31363)
5 years ago
Qi Li 59940cb383
[ROCM] update fluid operators for rocm (part8), test=develop (#31309)
5 years ago
tangwei12 5d7a8b05f8
fix sycn training error (#31357)
5 years ago
Qi Li ec72f5b235
fix ELU output for nan, test=develop (#31132)
5 years ago
Qi Li 65bcaeb004
[ROCM] update fluid operators for rocm (part5), test=develop (#31258)
5 years ago
YUNSHEN XIE 2111d912d4
Decrease threshold for failed ut retry (#30903)
5 years ago
Pei Yang 2e9e3fad15
add n-d input support for trt scale converter (#31316)
5 years ago
Shang Zhizhou 6404c43814
support trt serialize when load model from memory (#31342)
5 years ago
chentianyu03 a2c0b60401
remove wlist_temp in wlist.json (#31356)
5 years ago
Gradie d79fdc3d62
lamb_op_xpu;test=kunlun (#31012)
5 years ago
danleifeng d1075df2e8
topo and memory performance for heterps (#30440)
5 years ago
Qi Li 72d99c5dcd
[ROCM] update fluid operators for rocm (part4), test=develop (#31225)
5 years ago
cucuzg 91635de390
opt matmul and matmul_v2 on kunlun, *test=kunlun (#31326)
5 years ago
Wilber e20234094c
Fix xpu compile and cipher symbol problem. (#31271)
5 years ago
wuhuanzhou 30858d8974
fix compilation errors for missing brpc header files, test=develop (#31325)
5 years ago
石晓伟 625482f752
inference modification for custom operator, test=develop (#31312)
5 years ago
wuhuanzhou a13f1d6930
optimize unity build (#31119)
5 years ago
jiangcheng 8f4ac6b525
optimize topk op through limit SortTopK kernel entrance, test=develop (#30403)
5 years ago
alncat bfb8a64234
updated conv bn fuse pass to make it compatible with latest batch_norm op (#31272)
5 years ago
Bin Lu a37658daff
Update transforms.py (#31252)
5 years ago
Chen Weihang 5610c1717e
fix dtype unmatched (#31305)
5 years ago
Chen Long f8bdb90917
fix readme test=document_fix (#31308)
5 years ago
Qi Li 9b016c7cb7
[ROCM] update fluid operators for rocm (part2), test=develop (#31211)
5 years ago
niuliling123 2fd999d979
Optimized the adaptive_avg_pool2d op when output_size == 1 (#31197)
5 years ago
Zhou Wei aebf223478
fix test_check_abi (#31288)
5 years ago
石晓伟 1da3280660
inference modification for custom operator, test=develop (#31283)
5 years ago
Zhou Wei cc89120a2c
[Custom OP]add MSVC compile check on Windows (#31265)
5 years ago
Zhou Wei af9066e89c
[Custom OP]add PD_THROW and PD_CHECK for User Error message (#31253)
5 years ago
石晓伟 8c94d8cb4c
[Custom OP] change the user header file format, test=develop (#31274)
5 years ago
Jiabin Yang 038ce70d69
[Custom OP] Support stream set on Custom Op (#31257)
5 years ago
Aurelius84 1dd40870fc
[Dy2Stat] Fix eval_if_exist_else_none bug (#31261)
5 years ago