From c4ce9089670746866898c7bca16ed658e18193fc Mon Sep 17 00:00:00 2001 From: Christian Guinard <28689358+christiangnrd@users.noreply.github.com> Date: Wed, 5 Mar 2025 19:53:05 -0400 Subject: [PATCH] LowPrecArray tests --- test/lowprecarrays.jl | 25 +++++++++++++++++++++++++ test/runtests.jl | 1 + 2 files changed, 26 insertions(+) create mode 100644 test/lowprecarrays.jl diff --git a/test/lowprecarrays.jl b/test/lowprecarrays.jl new file mode 100644 index 0000000..afa3854 --- /dev/null +++ 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" \ No newline at end of file diff --git a/test/runtests.jl b/test/runtests.jl index 6579f5a..a97e714 100644 --- 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"