Switching to the correct OrtMemoryInfo name

egalli · Aug 7, 2024 · f5a94e2 · f5a94e2
1 parent 0265169
commit f5a94e2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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;