From f5a94e2d6696ca43f92e8f84fdce73df29db3751 Mon Sep 17 00:00:00 2001 From: Enrico Galli Date: Wed, 7 Aug 2024 14:55:13 -0700 Subject: [PATCH] Switching to the correct OrtMemoryInfo name --- onnxruntime/core/providers/webnn/allocator.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/onnxruntime/core/providers/webnn/allocator.h b/onnxruntime/core/providers/webnn/allocator.h index 9c3eff53fa842..6d9fd2c0542e2 100644 --- a/onnxruntime/core/providers/webnn/allocator.h +++ b/onnxruntime/core/providers/webnn/allocator.h @@ -15,7 +15,7 @@ namespace webnn { class WebNNBufferAllocator : public IAllocator { public: - WebNNBufferAllocator() : IAllocator(OrtMemoryInfo("WebNNMLBufferAllocator", OrtAllocatorType::OrtDeviceAllocator, OrtDevice(OrtDevice::GPU, OrtDevice::MemType::DEFAULT, 0), 0, OrtMemTypeDefault)) {} + WebNNBufferAllocator() : IAllocator(OrtMemoryInfo(WEBNN_BUFFER, OrtAllocatorType::OrtDeviceAllocator, OrtDevice(OrtDevice::GPU, OrtDevice::MemType::DEFAULT, 0), 0, OrtMemTypeDefault)) {} void* Alloc(size_t size) override;