LowPrecArray tests

JuliaMath · Mar 6, 2025 · c4ce908 · c4ce908
1 parent c557bc8
commit c4ce908
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 0 deletions.
diff --git a/test/lowprecarrays.jl b/test/lowprecarrays.jl
@@ -0,0 +1,25 @@
+@testset "lowprecarrays" begin
+    A = rand(Float32, 10,10)
+    lpA = LowPrecArray(A)
+
+    @test lpA isa LowPrecArray
+    @test size(A) == size(lpA)
+
+   lpA[3] = 4
+    @test lpA[3] == A[3]
+
+    a = rand(Float32)
+    ab = BFloat16s.ExpandingBFloat16(a)
+    b = rand(Float32)
+    bb = BFloat16s.ExpandingBFloat16(b)
+
+    @test ab isa BFloat16s.ExpandingBFloat16
+    @test ab.a isa BFloat16
+    @test ab * bb isa Float32
+
+    B = LowPrecArray(rand(Float32, 10,10))
+    C = LowPrecArray(rand(Float32, 10,10))
+
+    lpA .= B*C
+
+end # @testset "lowprecarrays"
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -243,5 +243,6 @@ end
 
 include("structure.jl")
 include("mathfuncs.jl")
+include("lowprecarrays.jl")
 
 end # @testset "BFloat16s"