correction

iitis · Sep 19, 2019 · 47e4eee · 47e4eee · kdomino · Sep 19, 2019
1 parent da46f27
commit 47e4eee
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/src/symten2mat.jl b/src/symten2mat.jl
@@ -49,18 +49,17 @@ Parallel implementation
 
 function computeblock_p(bm::SymmetricTensor{T, N}, i::Tuple{Int, Int}, dims::Tuple) where {T <: AbstractFloat, N}
   x = bm.bln^(N-2)
-  R = SharedArray(zeros(T, (x, makeblocksize(bm, i)...)))
+  @inbounds R = SharedArray(zeros(T, (makeblocksize(bm, i)..., x)))
   @sync @distributed for j in 1:x
     @inbounds k = Tuple(CartesianIndices(dims)[j])
     for k1 in 1:bm.bln
       @inbounds M1 = unfold(getblock(bm, (i[1],k1, k...)),1)
       @inbounds M2 = unfold(getblock(bm, (k1,i[2],k...)),2)
-      @inbounds R[j,:,:] += M1*transpose(M2)
+      @inbounds R[:,:,j] += M1*transpose(M2)
     end
   end
-  R = Array(R)
-  R = mapreduce(i -> R[i,:,:], +, 1:size(R,1))
-  return R
+  @inbounds R = Array(R)
+  return mapreduce(i -> R[:,:,i], +, 1:size(R,3))
 end
 """
   cum2mat(bm::SymmetricTensor{T, N}) where {T <: AbstractFloat, N}