From 1a1bb12611656d0c1c65489500c47bb5d4f8bc3c Mon Sep 17 00:00:00 2001 From: Avik Pal Date: Wed, 4 Sep 2024 16:07:41 -0400 Subject: [PATCH] fix: AMDGPU conv -> crosscor fix in testing --- test/shared_testsetup.jl | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/test/shared_testsetup.jl b/test/shared_testsetup.jl index bff60846a..b46e308ff 100644 --- a/test/shared_testsetup.jl +++ b/test/shared_testsetup.jl @@ -23,10 +23,10 @@ function get_default_rng(mode::String) end maybe_rewrite_to_crosscor(layer) = layer -function maybe_rewrite_to_crosscor(layer::Conv{N, use_bias, M}) where {N, use_bias, M} - return CrossCor{N, use_bias, M}( - layer.activation, layer.in_chs, layer.out_chs, layer.kernel_size, - layer.stride, layer.pad, layer.dilation, layer.init_weight, layer.init_bias) +function maybe_rewrite_to_crosscor(layer::Conv) + return CrossCor(layer.activation, layer.in_chs, layer.out_chs, layer.kernel_size, + layer.stride, layer.pad, layer.dilation, layer.groups, layer.init_weight, + layer.init_bias, layer.use_bias) end function maybe_rewrite_to_crosscor(mode, model)