Fix and test MPSGraph random

Author: christiangnrd

lib/mpsgraphs/random.jl

@@ -1,7 +1,7 @@
 # @objcwrapper immutable=false MPSGraphRandomOpDescriptor <: MPSGraphObject
 
-function MPSMatrixRandomOpDescriptor(distribution::MPSGraphRandomDistribution, dataType::MPSDataType)
-    desc = @objc [MPSMatrixRandomOpDescriptor descriptorWithDistribution:distribution::MPSGraphRandomDistribution
+function MPSGraphRandomOpDescriptor(distribution::MPSGraphRandomDistribution, dataType)
+    desc = @objc [MPSGraphRandomOpDescriptor descriptorWithDistribution:distribution::MPSGraphRandomDistribution
                     dataType:dataType::MPSDataType]::id{MPSGraphRandomOpDescriptor}
     obj = MPSGraphRandomOpDescriptor(desc)
     return obj

test/mpsgraphs/random.jl

@@ -0,0 +1,25 @@
+using BFloat16s
+
+if MPS.is_supported(device())
+
+using .MPSGraphs: MPSGraphRandomOpDescriptor, MPSGraphRandomDistributionNormal, MPSGraphRandomDistributionTruncatedNormal, MPSGraphRandomDistributionUniform
+@testset "MPSGraph random" begin
+    # determined by looking at the error message when trying to construct
+    #  an invalid distribution/type combination
+    for (dist, T) in [(MPSGraphRandomDistributionNormal, Float32),
+                      (MPSGraphRandomDistributionNormal, Float16),
+                      (MPSGraphRandomDistributionNormal, BFloat16),
+                      (MPSGraphRandomDistributionTruncatedNormal, Float32),
+                      (MPSGraphRandomDistributionTruncatedNormal, Float16),
+                      (MPSGraphRandomDistributionTruncatedNormal, BFloat16),
+                      (MPSGraphRandomDistributionUniform, Int64),
+                      (MPSGraphRandomDistributionUniform, Int32),
+                      (MPSGraphRandomDistributionUniform, Float32),
+                      (MPSGraphRandomDistributionUniform, Float16),
+                      (MPSGraphRandomDistributionUniform, BFloat16),
+                      ]
+        @test MPSGraphRandomOpDescriptor(MPSGraphRandomDistributionNormal, Float32) isa MPSGraphRandomOpDescriptor
+    end
+end
+
+end # MPS.is_supported(device())