JuliaGPU/CUDA.jl

Summing a vector is faster than summing a multidimensional array

オープン

#2,764 opened on 2025/04/28

 (1 件のコメント) (0 件のリアクション) (0 人の担当者)Julia (274 件のフォーク)batch import
cuda arraygood first issueperformance

Repository metrics

Stars
 (1,408 個のスター)
PR merge metrics
 (平均マージ 5d 5h) (30d で 16 merged PRs)

説明

I am not sure if this is an expected behavior, but it seems that sum(vec(A)) could be much faster than sum(A) when A is a multi-dimensional array. Please see https://discourse.julialang.org/t/summing-a-vector-is-faster-than-summing-a-multi-dimensional-array-of-the-same-length-using-cuda/116711 for an MWE. Although this post is from about a year ago, I can reproduce matching results with CUDA v5.7.3.

コントリビューターガイド