불필요한 할당

Question

는 다음과 같은 기능을 고려

x = rand(30,30,100000) 
b = rand(100000) 
@time mytest(x,b) 

elapsed time: 0.571765222 seconds (727837732 bytes allocated, 66.49% gc time) 

왜 그렇게 많은 메모리 및 지출 너무 많은 시간을 할당한다 : 나는 그것을 실행하면

function mytest(x, b) 
    y = zeros(x[:,:,1]) 
    for i in 1:length(b) 
     y += b[i] * x[:,:,i] 
    end 
    return y 
end 

, 나는 다음과 같은 수 가비지 수집을하고 있습니까? 코드는 유형이 안정해야하며 += 연산자가 재 할당을 수행하지 않을 것으로 예상됩니다. 그러나 두 개의 행렬을 추가 할 때마다 다시 할당하는 것으로 보입니다.

줄리아의 버그라고 생각합니까? 그리고 더 중요한 것은 어떻게 재 할당하지 않는 방식으로이 코드를 작성할 수 있습니까?

수정 : 고정 오타.

Answer 1

@ cd98은 할당 문제를 해결하는 3 중첩 루프 솔루션을 요청했지만 동등한 벡터화 버전보다 성능이 좋지 않을 것으로 예상합니다. 여기있다 :

function mytest(x, b) 
    d1, d2, d3 = size(x) 
    y = zeros(eltype(x), d1, d2) 
    for i in 1:d3 
     for j in 1:d2 
      for k in 1:d1 
       y[k,j] += b[i] * x[k,j,i] 
      end 
     end 
    end 
    return y 
end 

x = rand(30,30,100000) 
b = rand(100000) 
@time mytest(x,b) 
@time mytest(x,b) 

그리고 출력 :

elapsed time: 0.218220119 seconds (767172 bytes allocated) 
elapsed time: 0.197181799 seconds (7400 bytes allocated) 

Answer 2

Base.Cartesian를 사용할 수있는 가능성도있다 : using Base.Cartesian 후, 당신은

@nloops 3 i x begin 
    (@nref 2 y i) += b[i_3] * (@nref 3 x i) 
end 

이 같은 기본적에 확장하는 쓸 수 있습니다 짐의 대답처럼 루프.

Answer 3

는 (원본) 할당 문제가 해결되지 않습니다,하지만 난 단순히 @inbounds와 후자의 용액에 루프를 포장하여 1.8 배의 속도 향상에 대해 얻을 :

function mytest_inbounds(x, b) 
    d1, d2, d3 = size(x) 
    y = zeros(eltype(x), d1, d2) 
    @inbounds begin 
     for i in 1:d3 
      for j in 1:d2 
       for k in 1:d1 
        y[k,j] += b[i] * x[k,j,i] 
       end 
      end 
     end 
    end 
    return y 
end 

x = rand(30, 30, 100000) 
b = rand(100000) 
@time mytest(x, b) 
@time mytest(x, b) 
@time mytest_inbounds(x, b) 
@time mytest_inbounds(x, b) 

출력 :

elapsed time: 0.39144919 seconds (767212 bytes allocated) 
elapsed time: 0.353495867 seconds (7400 bytes allocated) 
elapsed time: 0.202614643 seconds (396972 bytes allocated) 
elapsed time: 0.193425902 seconds (7400 bytes allocated) 

또한, 여기에 관련 문제에 대한 좋은 토론의 많은 :

https://groups.google.com/forum/#!msg/julia-users/aYS_AvKqPCI/DyTiq4lKIAoJ