Commit graph

4950 commits

Author SHA1 Message Date
Ryan Hill
f4bb6f5c32 Oops, wrong CMakeLists 2021-05-12 18:22:34 -07:00
Ryan Hill
4ad3a41ff9 Patch through INcclService 2021-05-12 18:22:29 -07:00
Ryan Hill
3b2bd5f50d MPI Kernels part 2 2021-05-11 21:33:28 -07:00
Ryan Hill
66babe4ed3 MPI Kernels for Training 2021-05-11 20:16:42 -07:00
Ryan Hill
335562def6 Disable Nvtx temporarily 2021-05-11 16:17:39 -07:00
Ryan Hill
ff40ba0723 Still can't unload providers in python, alas. 2021-05-11 16:15:43 -07:00
Ryan Hill
330339caa4 Remove test_execution_provider from training build
Only enable python atexit on windows
Remove assert on provider library exit
2021-05-11 03:10:20 -07:00
Ryan Hill
fcb98063a2 Python crash on exit, possibly due to unloading of libraries. 2021-05-11 02:08:02 -07:00
Ryan Hill
c5b9d9cb3a Python typos 2021-05-10 22:51:56 -07:00
Ryan Hill
9b84d76342 Python typo 2021-05-10 22:20:19 -07:00
Ryan Hill
f5cd80afb2 Removed in wrong spot 2021-05-10 20:43:12 -07:00
Ryan Hill
02fbbdce35 Missing pass command 2021-05-10 20:13:50 -07:00
Ryan Hill
8e27b5d34a Fix test provider error 2021-05-10 20:10:24 -07:00
Ryan Hill
23c0011b42 training fixes 2021-05-10 19:50:28 -07:00
Ryan Hill
dfc8dab1b9 Last orttraining build fix... 2021-05-10 16:52:51 -07:00
Ryan Hill
149c037a4a For orttraining 2021-05-10 16:21:07 -07:00
Ryan Hill
a59cf4b050 Formatting 2021-05-10 16:21:01 -07:00
Ryan Hill
8c5d758038 Formatting 2021-05-10 16:20:56 -07:00
Ryan Hill
c419ff2487 Another java fix 2021-05-10 15:15:05 -07:00
Ryan Hill
6076ef965b Training fix 2021-05-10 15:09:35 -07:00
Ryan Hill
7bdf68ffa2 Revert "Remove LongformerAttentionBase workaround"
This reverts commit 648679b370.
2021-05-10 14:56:05 -07:00
Ryan Hill
567133235d Fix formatting 2021-05-10 14:09:17 -07:00
Ryan Hill
cd2c891a34 Fix nuphar 2 2021-05-10 14:09:11 -07:00
Ryan Hill
7cc57c11c8 Fix Nuphar? 2021-05-10 13:20:12 -07:00
Ryan Hill
12224c88b7 Java fix for CPU build 2021-05-10 13:03:17 -07:00
Ryan Hill
912b61e490 Java fix for non cuda case 2021-05-10 13:02:56 -07:00
Ryan Hill
49a8b5db0f Fix minimal build 2021-05-10 11:33:04 -07:00
Ryan Hill
a79227dd0f Spotless fix 2021-05-10 11:20:45 -07:00
Ryan Hill
11bec62921 Fix errors due to merge 2021-05-10 01:46:30 -07:00
Ryan Hill
1698c57fcc Merge with master 2021-05-10 00:41:07 -07:00
Ryan Hill
d308c57728 Code review feedback 2021-05-10 00:18:28 -07:00
Ryan Hill
64c0deeb7e Revert "Switch yml back to regular build pool"
This reverts commit be35fc2a5a.
2021-05-10 00:14:13 -07:00
Ryan Hill
be35fc2a5a Switch yml back to regular build pool 2021-05-09 10:48:58 -07:00
Hariharan Seshadri
4b691a5c0d
Add ability for memory arenas to "shrink" periodically (#7284) 2021-05-08 07:53:21 -07:00
Ye Wang
803837df63
Add 4dmask support for attention cuda kernel (#7591)
* checkin

* add 4dmask support in attention cuda op

* trim

* add comments

* fix build/test error

* review comments and add tests

* sync doc

* review comments

* minor change
2021-05-07 20:17:29 -07:00
Ryan Hill
e02270e640 Spotless fix 2021-05-07 19:18:23 -07:00
Ryan Hill
648679b370 Remove LongformerAttentionBase workaround 2021-05-07 18:11:06 -07:00
Ryan Hill
6ad01b9040 Java comments 2021-05-07 17:13:22 -07:00
Tianlei Wu
55c086b664
symbolic shape inference improvements for contrib ops (#7606)
* add EmbedLayerNormalization
* use onnx shape inference for Unsqueeze
* Fix type warning in Attention
2021-05-07 17:03:24 -07:00
Edward Chen
5a5fec0452
Fix logs getting skipped in single-line conditionals. (#7589)
Fix an issue where a log message got skipped.

A log call like this:
```
LOGS(...) << "message";
```
expands to something like this:
```
if (<output enabled>)
  logging::Capture(...).Stream() << "message";
```

This if statement without brackets is handy for logging arbitrary arguments with the `<<` operator. However, it has other drawbacks like possibly associating with a subsequent `else`.

```
if (cond)
  LOGS(...) << "a";
else
  <do something> // not run when !cond

// equivalently:
if (cond)
  if (<output enabled>)
    logging::Capture(...).Stream() << "a";
  else
    <do something> // not run when !cond
```

Updated the logging macros to handle this case by replacing `if (<enabled>) logging::Capture(...).Stream()` with `if (!<enabled>) {} else logging::Capture(...).Stream()`.

Thanks @tlh20 for the idea for the fix!
2021-05-07 15:40:47 -07:00
Derek Murray
e91bdbde20
Add myself to CODEOWNERS for ORTModule python code (#7453) 2021-05-07 15:35:45 -07:00
Changming Sun
41e370c2b3
Update protobuf to 3.16 (#7616) 2021-05-07 14:09:23 -07:00
Ryan Hill
981af7d085 Fix break 2021-05-07 13:59:26 -07:00
Ryan Hill
74b1016d3b Fix cleanup on exit 2021-05-07 13:16:17 -07:00
Tixxx
3c39fcc1fa
[js/web] port fixes for packed concat over to ort repo (#7605)
* port fixes for packed concat over to ort repo

* fix format
2021-05-07 13:04:53 -07:00
Yulong Wang
bdefc6c4d8
[js/web] support multi-thread for wasm backend (#7601) 2021-05-07 12:12:37 -07:00
stevenlix
8ab0deceed
Add DLA support to TensorRT EP (#7532)
* Add DLA to TensorRT EP, enable device_id options in pybind, fix cycledetection issue

* fix format

* remove unecessary passing by pointer

* fix issue
2021-05-07 10:31:42 -07:00
Scott McKay
9fc4116d51
Use ASSERT_STATUS_OK so the error message is output if there's a failure. (#7515) 2021-05-07 20:23:34 +10:00
Ryan Hill
9dfdafe6e0 Leave temp file around to load properly 2021-05-07 01:08:45 -07:00
Vincent Wang
0c91b643fe
Bugfix for Scatter and GatherElementsGrad (#7593)
* bugfix for scatter and gather elements grad

* resolve comments
2021-05-07 14:02:26 +08:00