onnxruntime
a4902ee6 - [CUDA][ROCm] Allow allocating ScratchBuffer from TuningContext (#17028)

Commit

2 years ago

[CUDA][ROCm] Allow allocating ScratchBuffer from TuningContext (#17028) By switching to ort native stream, we can allocate scratch buffer directly from tuning context.

References

#17028 - [CUDA][ROCm] Allow allocating ScratchBuffer from TuningContext

Author

cloudhan

Parents

6e6f582e

onnxruntime a4902ee6 - [CUDA][ROCm] Allow allocating ScratchBuffer from TuningContext (#17028)

onnxruntime
a4902ee6 - [CUDA][ROCm] Allow allocating ScratchBuffer from TuningContext (#17028)