Skip to content

Commit 2d11083

Browse files
committed
Add test for conv_bias_act
1 parent f4539bf commit 2d11083

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

ext/NNlibCUDA/test/conv.jl

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,12 @@ using NNlib: DenseConvDims
88
@test ∇conv_data(c, b, cdims) collect(∇conv_data(dc, db, cdims))
99
@test ∇conv_filter(a, c, cdims) collect(∇conv_filter(da, dc, cdims))
1010

11+
# Test Conv Bias Activation
12+
bias = rand(Float64, 1, 1, 4, 1)
13+
dbias = CuArray(bias)
14+
@test conv_bias_act(a, b, cdims, bias, NNlib.relu) collect(conv_bias_act(da, db, cdims, dbias, NNlib.relu))
15+
@test conv_bias_act(a, b, cdims, bias, identity) collect(conv_bias_act(da, db, cdims, dbias, identity))
16+
1117
# Test for agreement between CPU NNlib and CuDNN versions, across a variety of kwargs
1218
options = Dict{Any, Any}.((
1319
(), (:dilation => 2), (:flipkernel => true), (:stride => 2),

0 commit comments

Comments
 (0)