saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-07 00:13:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
onnxruntime/onnxruntime/python
History
raviskolli 5d759e182b
Allocate external Rocm allocator via PyBind (#7148) 
* Enabled rocm support for graph transformations

* Support for external Hip allocator

* Added const_cast to reinterpret_cast to fix compiler issue

* Another crack at fixing the compile error

* More compilation fixes

* Added compilation flags to load_inline extension

* Added ROCM, ROCM_PINNED constants

* Changes to address PR comments

* Changed gpu identifier from ROCM to CUDA

* Added HIP compilation flag for torch inline functions

* Fixed a typo in header allocator string formatting

* Fix for runtime error with external_cuda_allocator

* Removed cuda/rocm specific code paths for allocators

* More name changes to generic gpu from rocm/cuda

* Removed duplicate allocator creation

* Rename cuda_external_ config options as gpu_external_

* Rename hip_mem_limit to gpu_mem_limit

* Rename cuda_mem_limit to gpu_mem_limit
2021-04-06 15:23:51 -07:00
..
backend fix py packaging pipeline (#5038) 2020-09-03 09:32:10 -07:00
datasets replace invalid sample (#4567) 2020-07-21 23:51:17 -07:00
dlpack Address comments from ORTModule master merge (#7101) 2021-03-26 16:26:42 -07:00
tools Mem trt (#6868) 2021-04-05 22:16:12 -07:00
training Enable running PEP8 on python scripts using flake8 (#3928) 2020-05-15 07:15:06 +10:00
__init__.py Enable running PEP8 on python scripts using flake8 (#3928) 2020-05-15 07:15:06 +10:00
_ld_preload.py Enable running PEP8 on python scripts using flake8 (#3928) 2020-05-15 07:15:06 +10:00
_pybind_state.py Add cupti path for python gpu packaging pipeline (#7200) 2021-04-02 12:12:46 -07:00
exported_symbols.lst
onnxruntime_inference_collection.py Enable NHWC transformer when generating ORT format model (#7126) 2021-03-29 18:39:48 +10:00
onnxruntime_pybind_exceptions.h
onnxruntime_pybind_mlvalue.cc Support sharing of initializers between session via the Python API (#5407) 2020-10-09 20:26:28 -07:00
onnxruntime_pybind_mlvalue.h Support sharing of initializers between session via the Python API (#5407) 2020-10-09 20:26:28 -07:00
onnxruntime_pybind_state.cc Allocate external Rocm allocator via PyBind (#7148) 2021-04-06 15:23:51 -07:00
onnxruntime_pybind_state_common.cc Enable passing initial optimizer state while creating training session (#5869) 2020-12-08 21:20:51 -05:00
onnxruntime_pybind_state_common.h Enable NHWC transformer when generating ORT format model (#7126) 2021-03-29 18:39:48 +10:00
onnxruntime_validation.py Enable running PEP8 on python scripts using flake8 (#3928) 2020-05-15 07:15:06 +10:00
pybind.def
version_script.lds