diff options
Diffstat (limited to 'src/PackWeightMatrixForGConv.cc')
-rw-r--r-- | src/PackWeightMatrixForGConv.cc | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/PackWeightMatrixForGConv.cc b/src/PackWeightMatrixForGConv.cc index 5870fa5..0fb0e2c 100644 --- a/src/PackWeightMatrixForGConv.cc +++ b/src/PackWeightMatrixForGConv.cc @@ -19,7 +19,7 @@ PackWeightMatrixForGConv<T, accT, SPATIAL_DIM>::PackWeightMatrixForGConv( const T* sdata, T* pdata) : trans_(trans), conv_param_(conv_param), sdata_(sdata) { - static_assert(SPATIAL_DIM == 2, "3D conv not supported yet"); + assert(SPATIAL_DIM == 2 && "3D conv not supported yet"); if (!pdata) { bufAllocatedHere_ = true; @@ -111,4 +111,6 @@ void PackWeightMatrixForGConv<T, accT, SPATIAL_DIM>::pack() { template class PackWeightMatrixForGConv<int8_t, int32_t, 2>; template class PackWeightMatrixForGConv<int8_t, int16_t, 2>; +template class PackWeightMatrixForGConv<int8_t, int32_t, 3>; +template class PackWeightMatrixForGConv<int8_t, int16_t, 3>; } // namespace fbgemm |