Skip to content

Commit daa862a

Browse files
author
Dhairya Gandhi
committed
obscuring conv_nnpack for CI
1 parent b8fb492 commit daa862a

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

test/conv.jl

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -274,9 +274,13 @@ conv_answer_dict = Dict(
274274
# A "drop channels and batch dimension" helper
275275
ddims(x) = dropdims(x, dims=(rank+1, rank+2))
276276

277-
for conv in (NNlib.conv, NNlib.conv_im2col, NNlib.conv_direct, NNlib.conv_nnpack)
278-
if conv == NNlib.conv_nnpack && !NNlib.nnpack_supported_operation(DenseConvDims(x, w))
279-
continue
277+
convs = [NNlib.conv, NNlib.conv_im2col, NNlib.conv_direct,]
278+
NNlib.is_nnpack_available() && push!(convs, NNlib.conv_nnpack)
279+
for conv in convs
280+
if NNlib.is_nnpack_available()
281+
if conv == NNlib.conv_nnpack && !NNlib.nnpack_supported_operation(DenseConvDims(x, w))
282+
continue
283+
end
280284
end
281285
@testset "$(conv)" begin
282286
cdims = DenseConvDims(x, w)

0 commit comments

Comments
 (0)