Skip to content

Commit c9327e6

Browse files
committed
Fix tests on Metal
1 parent 089d19d commit c9327e6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

NDTensors/src/lib/BlockSparseArrays/test/test_basics.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -298,7 +298,7 @@ using .NDTensorsTestUtils: devices_list, is_supported_eltype
298298
@test nstored(b) == 2 * 4 + 3 * 3
299299

300300
a = dev(BlockSparseArray{elt}([1, 1, 1], [1, 2, 3], [2, 2, 1], [1, 2, 1]))
301-
a[Block(3, 2, 2, 3)] = dev(randn(1, 2, 2, 1))
301+
a[Block(3, 2, 2, 3)] = dev(randn(elt, 1, 2, 2, 1))
302302
perm = (2, 3, 4, 1)
303303
for b in (PermutedDimsArray(a, perm), permutedims(a, perm))
304304
@test Array(b) == permutedims(Array(a), perm)

0 commit comments

Comments
 (0)