Welcome to mirror list, hosted at ThFree Co, Russian Federation.

github.com/marian-nmt/FBGEMM.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
Diffstat (limited to 'src/PackWeightsForConv.cc')
-rw-r--r--src/PackWeightsForConv.cc3
1 files changed, 2 insertions, 1 deletions
diff --git a/src/PackWeightsForConv.cc b/src/PackWeightsForConv.cc
index e16843c..085adc0 100644
--- a/src/PackWeightsForConv.cc
+++ b/src/PackWeightsForConv.cc
@@ -13,7 +13,8 @@ template <int SPATIAL_DIM, typename T, typename accT>
PackWeightsForConv<SPATIAL_DIM, T, accT>::PackWeightsForConv(
const conv_param_t<SPATIAL_DIM>& conv_p,
const T* sdata,
- const BlockingFactors* blocking_params) {
+ const BlockingFactors* blocking_params)
+ : conv_param_(conv_p) {
static_assert(
SPATIAL_DIM == 2 || SPATIAL_DIM == 3,
"Only 2D and 3D convolutions are supported");