Skip to content

Commit 5108ab2

Browse files
bors[bot]ToucheSir
andauthored
Merge #1836
1836: Try using latest cu(DNN) binaries r=ToucheSir a=ToucheSir Possible alternative to #1822. Co-authored-by: Brian Chen <ToucheSir@users.noreply.github.com>
2 parents 79dbbd6 + 6e3ca67 commit 5108ab2

File tree

2 files changed

+3
-5
lines changed

2 files changed

+3
-5
lines changed

.buildkite/pipeline.yml

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,8 @@ steps:
1717
agents:
1818
queue: "juliagpu"
1919
cuda: "*"
20+
env:
21+
JULIA_CUDA_USE_BINARYBUILDER: "true"
2022
timeout_in_minutes: 60
2123

2224
# - label: "GPU nightly"

test/cuda/layers.jl

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -45,11 +45,7 @@ function gpu_gradtest(name::String, layers::Vector, x_cpu = nothing, args...; te
4545

4646
# test
4747
if test_cpu
48-
if VERSION >= v"1.7" && layer === GroupedConvTranspose && args[end] == selu
49-
@test_broken y_gpu y_cpu rtol=1f-3 atol=1f-3
50-
else
51-
@test y_gpu y_cpu rtol=1f-3 atol=1f-3
52-
end
48+
@test y_gpu y_cpu rtol=1f-3 atol=1f-3
5349
if isnothing(xg_cpu)
5450
@test isnothing(xg_gpu)
5551
else

0 commit comments

Comments
 (0)