File tree Expand file tree Collapse file tree 1 file changed +7
-3
lines changed Expand file tree Collapse file tree 1 file changed +7
-3
lines changed Original file line number Diff line number Diff line change @@ -274,9 +274,13 @@ conv_answer_dict = Dict(
274
274
# A "drop channels and batch dimension" helper
275
275
ddims (x) = dropdims (x, dims= (rank+ 1 , rank+ 2 ))
276
276
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
280
284
end
281
285
@testset " $(conv) " begin
282
286
cdims = DenseConvDims (x, w)
You can’t perform that action at this time.
0 commit comments