Skip to content

Commit 8e8eca4

Browse files
committed
Fix format
1 parent 1e4a01a commit 8e8eca4

1 file changed

Lines changed: 2 additions & 3 deletions

File tree

perf/arraydiff_gpu.jl

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -168,7 +168,7 @@ end
168168
# -------------------------------------------------------------------------
169169
# One (h, d, n) sweep.
170170
# -------------------------------------------------------------------------
171-
function run_one(; h::Int, d::Int = 13, n::Int = 178, rtol::Float32 = 1f-3)
171+
function run_one(; h::Int, d::Int = 13, n::Int = 178, rtol::Float32 = 1.0f-3)
172172
println("\n" * "="^72)
173173
@printf "h = %d, d = %d, n = %d (Float32)\n" h d n
174174
println("="^72)
@@ -198,8 +198,7 @@ function run_one(; h::Int, d::Int = 13, n::Int = 178, rtol::Float32 = 1f-3)
198198
# Hand-CUDA preallocated. Same arithmetic, but every intermediate is
199199
# owned by `hand_buf` and reused across calls.
200200
hand_buf = HandPrealloc{T}(h, d, n, out_dim)
201-
grad_prealloc =
202-
Array(reverse_diff_prealloc!(hand_buf, W1g, W2g, Xg, yg))
201+
grad_prealloc = Array(reverse_diff_prealloc!(hand_buf, W1g, W2g, Xg, yg))
203202
CUDA.synchronize()
204203

205204
# ArrayDiff CPU.

0 commit comments

Comments
 (0)