[METAL] Use CFBridgeRetain for retaining the allocated resource (apac…

…he#6393)
neo-ai · Sep 18, 2020 · 8cee105 · 8cee105
1 parent c29d491
commit 8cee105
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/runtime/metal/metal_device_api.mm b/src/runtime/metal/metal_device_api.mm
@@ -1,3 +1,4 @@
+
 /*
  * Licensed to the Apache Software Foundation (ASF) under one
  * or more contributor license agreements.  See the NOTICE file
@@ -159,7 +160,7 @@ int GetWarpSize(id<MTLDevice> dev) {
   */
   id<MTLBuffer> buf = [dev newBufferWithLength:nbytes options:storage_mode];
   CHECK(buf != nil);
-  return (__bridge void*)([buf retain]);
+  return (void*)(CFBridgingRetain(buf));
 }
 
 void MetalWorkspace::FreeDataSpace(TVMContext ctx, void* ptr) {