Skip to content

Commit c86b0ce

Browse files
Katharine HyattKatharine Hyatt
authored andcommitted
Restore Lukas' fixes
1 parent bd3651a commit c86b0ce

2 files changed

Lines changed: 3 additions & 5 deletions

File tree

ext/TensorKitAMDGPUExt/TensorKitAMDGPUExt.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,13 +6,13 @@ import AMDGPU: rand as rocrand, rand! as rocrand!, randn as rocrandn, randn! as
66

77
using TensorKit
88
using TensorKit.Factorizations
9-
using TensorKit.Strided
9+
using Strided
10+
using MatrixAlgebraKit
1011
using MatrixAlgebraKit: AbstractAlgorithm
1112
using TensorKit: SectorDict, tensormaptype, scalar, similarstoragetype, AdjointTensorMap, scalartype, project_symmetric_and_check
1213
import TensorKit: randisometry
1314
using Base: rand, randn
1415

15-
using TensorKit: MatrixAlgebraKit
1616

1717
using Random
1818

ext/TensorKitAMDGPUExt/roctensormap.jl

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -86,9 +86,7 @@ for randfun in (:rocrand, :rocrandn)
8686
end
8787

8888
function $randfun!(rng::Random.AbstractRNG, t::ROCTensorMap)
89-
for (_, b) in blocks(t)
90-
$randfun!(rng, b)
91-
end
89+
$randfun!(rng, b.data)
9290
return t
9391
end
9492
end

0 commit comments

Comments
 (0)