@@ -4,18 +4,24 @@ using .MPSGraphs: MPSGraphRandomOpDescriptor, MPSGraphRandomDistributionNormal,
44@testset " MPSGraph random" begin
55 # determined by looking at the error message when trying to construct
66 # an invalid distribution/type combination
7- for (dist, T) in [(MPSGraphRandomDistributionNormal, Float32),
7+ OP_TYPES = [(MPSGraphRandomDistributionNormal, Float32),
88 (MPSGraphRandomDistributionNormal, Float16),
9- (MPSGraphRandomDistributionNormal, BFloat16),
109 (MPSGraphRandomDistributionTruncatedNormal, Float32),
1110 (MPSGraphRandomDistributionTruncatedNormal, Float16),
12- (MPSGraphRandomDistributionTruncatedNormal, BFloat16),
1311 (MPSGraphRandomDistributionUniform, Int64),
1412 (MPSGraphRandomDistributionUniform, Int32),
1513 (MPSGraphRandomDistributionUniform, Float32),
1614 (MPSGraphRandomDistributionUniform, Float16),
17- (MPSGraphRandomDistributionUniform, BFloat16),
1815 ]
16+
17+ if Metal. macos_version () >= v " 14"
18+ append! (OP_TYPES, [(MPSGraphRandomDistributionNormal, BFloat16),
19+ (MPSGraphRandomDistributionTruncatedNormal, BFloat16),
20+ (MPSGraphRandomDistributionUniform, BFloat16),]
21+ )
22+ end
23+
24+ for (dist, T) in OP_TYPES
1925 @test MPSGraphRandomOpDescriptor (dist, T) isa MPSGraphRandomOpDescriptor
2026 end
2127end
0 commit comments