다음 글 참고 : https://github.com/google/jax/issues/5733
Jax throws internal XLA error for sufficiently large batched outer product on GPU · Issue #5733 · google/jax
Code: import jax.numpy as jnp A = jnp.ones((4, 1024*1024, 1)) B = jnp.ones((4, 1024*1024, 1)) tot = 0 def f(A, B): return A[:,:,None,:] * B[:,None, :, :] f(A, B) throws /usr/local/lib/python3.6/dis...
github.com
너무 많은 메모리를 요구해서 발생하는 에러.
실제로 함수 돌릴 때 그렇게 큰 메모리를 요구한다고 느끼지는 않았는데 의외다.
'프로그래밍 Programming' 카테고리의 다른 글
| [Gym] Gymnasium custom environment (0) | 2024.06.02 |
|---|---|
| [Python] 각종 라이브러리 버전체크 방법 (0) | 2024.05.16 |
| [Julia] 변수 Variables (0) | 2024.04.30 |
| [Julia] Julia 외부 라이브러리 간단 사용법 (0) | 2024.03.20 |
| [Julia] Julia 프로그래밍 공부자료 (0) | 2024.02.06 |