More API

Author: christiangnrd

lib/mpsgraphs/core.jl

@@ -27,6 +27,14 @@ function MPSGraphDevice(device::MTLDevice)
     MPSGraphDevice(obj)
 end
 
+# @objcwrapper MPSGraphExecutionDescriptor <: MPSGraphObject
+
 function MPSGraphExecutionDescriptor()
     MPSGraphExecutionDescriptor(@objc [MPSGraphExecutionDescriptor new]::id{MPSGraphExecutionDescriptor})
 end
+
+# @objcwrapper MPSGraphCompilationDescriptor <: MPSGraphObject
+
+function MPSGraphCompilationDescriptor()
+    MPSGraphCompilationDescriptor(@objc [MPSGraphCompilationDescriptor new]::id{MPSGraphCompilationDescriptor})
+end

lib/mpsgraphs/operations.jl

@@ -64,3 +64,13 @@ function identityWithTensor(graph::MPSGraph, tensor::MPSGraphTensor, name = "ide
                                 name:name::id{NSString}]::id{MPSGraphTensor}
     MPSGraphTensor(obj)
 end
+
+"""
+    dump_graph(graph::MPSGraph)
+
+Dumps the `graph`.
+
+!!! warning
+    This function is undocumented from Apple so it may stop working at any time.
+"""
+dump_graph(graph::MPSGraph) = @objc [graph::id{MPSGraph} dump]::Nothing