Skip to content

Commit 96812f2

Browse files
committed
Fix CUDA benchmark init
1 parent 64724d1 commit 96812f2

File tree

2 files changed

+5
-3
lines changed

2 files changed

+5
-3
lines changed

.gitignore

+2
Original file line numberDiff line numberDiff line change
@@ -5,3 +5,5 @@
55
.*sw?
66
Manifest.toml
77
*.jld
8+
*.jls
9+
*.dot

benchmarks/benchmark.jl

+3-3
Original file line numberDiff line numberDiff line change
@@ -150,9 +150,9 @@ function nmf_suite(; dagger, accel, kwargs...)
150150
nnmf($X[], $W[], $H[])
151151
end setup=begin
152152
if $accel == "cuda"
153-
$X[] = CUDA.rand((Float32, $nrow, $ncol))
154-
$W[] = CUDA.rand((Float32, $nrow, $nfeatures))
155-
$H[] = CUDA.rand((Float32, $nfeatures, $ncol))
153+
$X[] = CUDA.rand(Float32, $nrow, $ncol)
154+
$W[] = CUDA.rand(Float32, $nrow, $nfeatures)
155+
$H[] = CUDA.rand(Float32, $nfeatures, $ncol)
156156
elseif $accel == "amdgpu"
157157
$X[] = ROCArray(rand(Float32, $nrow, $ncol))
158158
$W[] = ROCArray(rand(Float32, $nrow, $nfeatures))

0 commit comments

Comments
 (0)