From c9327e65a688eaf47a56369e91a88fff6779e376 Mon Sep 17 00:00:00 2001 From: mtfishman Date: Sat, 9 Nov 2024 11:45:33 -0500 Subject: [PATCH] Fix tests on Metal --- NDTensors/src/lib/BlockSparseArrays/test/test_basics.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/NDTensors/src/lib/BlockSparseArrays/test/test_basics.jl b/NDTensors/src/lib/BlockSparseArrays/test/test_basics.jl index 3cfaeded6d..57de184afe 100644 --- a/NDTensors/src/lib/BlockSparseArrays/test/test_basics.jl +++ b/NDTensors/src/lib/BlockSparseArrays/test/test_basics.jl @@ -298,7 +298,7 @@ using .NDTensorsTestUtils: devices_list, is_supported_eltype @test nstored(b) == 2 * 4 + 3 * 3 a = dev(BlockSparseArray{elt}([1, 1, 1], [1, 2, 3], [2, 2, 1], [1, 2, 1])) - a[Block(3, 2, 2, 3)] = dev(randn(1, 2, 2, 1)) + a[Block(3, 2, 2, 3)] = dev(randn(elt, 1, 2, 2, 1)) perm = (2, 3, 4, 1) for b in (PermutedDimsArray(a, perm), permutedims(a, perm)) @test Array(b) == permutedims(Array(a), perm)