Tensor.Creation: Added IntPtr.Zero check for FreeHGlobalMemory

Nucs · Nucs · commit 1d6f3de37da9 · 2019-08-30T21:20:56.000+03:00
diff --git a/src/TensorFlowNET.Core/Tensors/Tensor.Creation.cs b/src/TensorFlowNET.Core/Tensors/Tensor.Creation.cs
@@ -664,7 +664,7 @@ protected unsafe IntPtr CreateTensorWithoutCopying(TF_DataType dt, long[] shape,
         [MonoPInvokeCallback(typeof(Deallocator))]
         internal static void FreeHGlobalMemory(IntPtr dataPtr, IntPtr len, ref DeallocatorArgs args)
         {
-            if (args.deallocator_called)
+            if (args.deallocator_called || dataPtr == IntPtr.Zero)
                 return;
 
             // NumSharp will dispose

Original file line number	Diff line number	Diff line change
`@@ -664,7 +664,7 @@ protected unsafe IntPtr CreateTensorWithoutCopying(TF_DataType dt, long[] shape,`
`664`	`664`	`[MonoPInvokeCallback(typeof(Deallocator))]`
`665`	`665`	`internal static void FreeHGlobalMemory(IntPtr dataPtr, IntPtr len, ref DeallocatorArgs args)`
`666`	`666`	`{`
`667`		`- if (args.deallocator_called)`
	`667`	`+ if (args.deallocator_called \|\| dataPtr == IntPtr.Zero)`
`668`	`668`	`return;`
`669`	`669`
`670`	`670`	`// NumSharp will dispose`