onnxruntime/tools
Weixing Zhang ca9b3f18e9
Explicitly pass cuda stream to thrust function rather than use cuda default stream implicitly (#7414)
* Pass cuda stream to thrust function to not use default stream.

In the commit 299ace0, ORT has been changed to not use cuda default stream.

* update amd_hipify.py

* remove un-necessary stream sync

Co-authored-by: Weixing Zhang <wezhan@microsoft.com>
2021-04-25 01:18:56 -07:00
..
ci_build Explicitly pass cuda stream to thrust function rather than use cuda default stream implicitly (#7414) 2021-04-25 01:18:56 -07:00
doc Add auto doc gen for ORTModule API during CI build (#7046) 2021-03-22 10:20:33 -07:00
nuget Only set _native folder for Microsoft.AI.MachineLearning package (#6939) 2021-03-08 15:27:11 -08:00
perf_util Update mysql-connector-java (#5802) 2020-11-16 14:09:14 -08:00
python [JS] refactor Javascript/Typescript libraries in ONNX Runtime (#7308) 2021-04-16 01:33:10 -07:00
test Add Trilu custom op (#4537) 2020-08-17 14:42:26 -07:00