Align context virtuals (#18396)

Deprecate ROCM context virtual function, to align with CUDA.

Co-authored-by: Randy Shuai <rashuai@microsoft.com>
This commit is contained in:
RandySheriffH 2023-11-10 18:41:37 -08:00 committed by GitHub
parent d87d480857
commit 646f77a94b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -18,7 +18,7 @@ struct RocmContext : public CustomOpContext {
miopenHandle_t miopen_handle = {};
rocblas_handle rblas_handle = {};
void Init(const OrtKernelContext& kernel_ctx) override {
void Init(const OrtKernelContext& kernel_ctx) {
const auto& ort_api = Ort::GetApi();
void* resource = {};
OrtStatus* status = nullptr;