Commit graph

4717 commits

Author SHA1 Message Date
Ryan Hill
96fa0845a2 Diagnostics 2021-04-23 13:39:49 -07:00
Ryan Hill
5c6910ed9c Fix memory cleanup on unload 2021-04-22 23:07:56 -07:00
Ryan Hill
a36dd5d574 Fix some setup issues 2021-04-22 17:56:03 -07:00
Ryan Hill
0a20e40af1 No-op cuda provider creation 2021-04-22 14:44:44 -07:00
Ryan Hill
f2d14cd74e Make GetProviderInfo_CUDA throw 2021-04-22 14:43:11 -07:00
Ryan Hill
4b8fba4f26 Test not adding cuda provider to session 2021-04-21 23:39:38 -07:00
Ryan Hill
4cf9e0b9a3 Remove more header dependencies 2021-04-21 23:35:31 -07:00
Ryan Hill
9f1c2ed767 Restore cuda in test 2021-04-21 20:28:03 -07:00
Ryan Hill
e74de85a53 Merge branch 'master' of https://github.com/microsoft/onnxruntime into ryanunderhill/cuda_shared 2021-04-21 19:55:50 -07:00
Ryan Hill
663ebee6b9 Cleanup 2021-04-21 19:55:40 -07:00
Thiago Crepaldi
771a6d235b
Fix IsContiguousTensor check on backend (#7391) 2021-04-21 17:01:17 -07:00
Changming Sun
afa7b23609
Update docs/ContribOperators.md and the script that generates it. (#7399) 2021-04-21 16:20:56 -07:00
Brian Popow
1bbe538379 Update references 2021-04-21 13:36:10 -07:00
Brian Popow
aa1ce726aa Remove unnecessary encoding step 2021-04-21 13:36:10 -07:00
Changming Sun
65b2b87f83
Update CI build docker images (#7386)
Update CI build docker images: delete ubuntu 16.04 support.
2021-04-21 13:18:34 -07:00
raviskolli
09313d9e1f
Added GreaterOrEqual and LessOrEqual Ops to RocmEP (#7398)
* Added GreaterOrEqual and LessOrEqual Ops to Rocm EP
2021-04-21 11:44:24 -07:00
Changming Sun
b4cfa88bf7
Update protobuf to the latest version (#7396) 2021-04-21 10:30:06 -07:00
Changming Sun
243713c464
Upload detailed code coverage result to azure blob storage (#7392) 2021-04-21 08:24:44 -07:00
Sherlock
16ca7677e6
Relax ConvGrad Test tol (#7393)
Co-authored-by: Sherlock Huang <bahuang@OrtTrainingDev3.af05slrtruoetgaxwwjv5nsq5e.px.internal.cloudapp.net>
2021-04-21 08:06:00 -07:00
Ryan Hill
7bf0090206 Test disabling cast 2021-04-21 01:50:59 -07:00
Ryan Hill
95fa78df38 Fix pybind 2021-04-20 22:40:38 -07:00
Changming Sun
b5493d724c
Update rnn_helpers.cc: add #ifdef to DumpMatrixImpl (#7389) 2021-04-20 22:11:38 -07:00
Ryan Hill
29e18e7dd5 Fix cuda memory leak 2021-04-20 21:10:38 -07:00
Hariharan Seshadri
7b11283af0
Add ability to allocate initialized tensor memory from non-arena memory (#7267) 2021-04-20 20:27:48 -07:00
Thiago Crepaldi
8421124344
Add support to **kwargs in ORTModule forward() method (#7360) 2021-04-20 16:21:52 -07:00
ashbhandare
76cc118dbe
Gemm transpose fusion (#7306)
* Gemm transpose fusion

* Correct rewrite rule effect

* Add to inference transforms to trigger on gradient graph
2021-04-20 09:35:05 -07:00
Xiaoyu Liu
913ea8264b
GPT2 with one step beam search (#7163)
* beam search refactoring checkin
* add factory class and deduplicate code
* one step beam search works on gpu

Co-authored-by: Xiaoyu Liu <xiaoyu@xiaoyu-VM.z4vh1dzj5eoevgybsksdpz2izh.jx.internal.cloudapp.net>
2021-04-20 06:23:52 -07:00
mindest
1a3ddf0714
Add gradient registration and tests for Min/Max (#7217)
* Add gradient registration and tests for Min/Max

* Add helper function for min/max grad test

* limit Min/Max Grad to accept at most two inputs; modify test case accordingly

* resolve merge error
2021-04-20 18:14:31 +08:00
Ryan Hill
263ca76307 Fix template syntax for linux 2021-04-20 02:47:14 -07:00
Ryan Hill
4e8eab65ef Remove extra qualifiers 2021-04-20 02:26:00 -07:00
Ryan Hill
21a734861f Fix control flow ops to use common base class with core code 2021-04-20 02:05:41 -07:00
Sherlock
ce7ff27bac
Fix perf issue in Conv CUDA kernel (#7348)
* Fix perf issue in  Conv CUDA kernel

* Read avaiable memory from device

* assuming 10% fragmentation

Co-authored-by: Sherlock Huang <bahuang@OrtTrainingDev3.af05slrtruoetgaxwwjv5nsq5e.px.internal.cloudapp.net>
2021-04-19 23:37:05 -07:00
ashbhandare
ac346a1b90
Modify SimplifiedLayerNormFusion to allow fusion in the presence of Casts optionally (#7352)
* LN transform partial changes

* LN transform fix

* Make transform optional, remove unnecessary code

* Fix windows build

* review comment, windows CI fix

* review comments
2021-04-19 19:59:23 -07:00
Ryan Hill
d9d2c3d538 Build test 2021-04-19 18:58:29 -07:00
ytaous
7abe1fd392
Identity elimination with graph output (#7312)
* Identity removal

* fix build

* fix build

* fix build

* fix builld

* UTs

* fix UT

* fix UTs

* per comments

* fix UTs

* fix UTs

* per comments

Co-authored-by: Ethan Tao <ettao@OrtTrainingDev4.af05slrtruoetgaxwwjv5nsq5e.px.internal.cloudapp.net>
2021-04-19 16:36:35 -07:00
Sheil Kumar
265db2ad96
Fix Microsoft.AI.MachineLearning .NET5 publishing and C# Store Release build (#7373)
* fix .net publishing

* make experimental api build with microsoft.ai.machinelearning.idl import

Co-authored-by: Sheil Kumar <sheilk@microsoft.com>
2021-04-19 15:36:43 -07:00
Ryan Hill
7c3855d655 Fix linux link error 2021-04-19 15:34:07 -07:00
Ryan Hill
03bdd1cf62 Another build test 2021-04-19 14:26:38 -07:00
satyajandhyala
bb1e417da0
Add logging support to Cast Propagation transformation from python (#7353)
* Fixes needed to PropagateCast transformation.

* Added number of passes to the logs.

* Added logging support to OrtModuleGraphBuilder.

* Added new testcases.

* Added NodeArgToConsumerMap
2021-04-19 12:14:30 -07:00
Ryan Hill
8159924918 Another test 2021-04-17 01:12:36 -07:00
Ryan Hill
81ad74d19a Linker test 2021-04-16 23:36:40 -07:00
Ryan Hill
8344a1cd3e Linker test 2021-04-16 22:50:50 -07:00
Ryan Hill
7f6ad2e825 Another linker test 2021-04-16 22:50:32 -07:00
Ryan Hill
9cfeff7fdc Test link error 2021-04-16 22:28:10 -07:00
Ryan Hill
5173d1c721 Fix unused parameter warning 2021-04-16 19:45:37 -07:00
Ryan Hill
d254c31756 Fix Eigen build error 2021-04-16 19:21:54 -07:00
Ryan Hill
dcafa85af2 Fix linux build 2021-04-16 18:04:40 -07:00
M. Zeeshan Siddiqui
6dda1e0681
Flag for tensor memory re-use in allocation planner. (#7359) 2021-04-16 17:53:25 -07:00
Ryan Hill
2f7bb27fa0 Test fix for linux 2021-04-16 17:33:42 -07:00
Guoyu Wang
96cdc65d57
Fix android CI failure after gradle updated to 7.0 (#7364)
* Fix android ci failure after gradle updated to 7.0

* minor update
2021-04-16 15:28:28 -07:00