From feca34d3d0b52527a1caa53186bc2ebf72d227c2 Mon Sep 17 00:00:00 2001
From: Daya Khudia <dskhudia@fb.com>
Date: Mon, 15 Jul 2019 17:34:48 -0700
Subject: Add functions needed for unpacking in PackWeightsForConv (#106)

Summary:
Pull Request resolved: https://github.com/pytorch/FBGEMM/pull/106

The values returned by these functions is needed while unpacking weights.

Reviewed By: jianyuh

Differential Revision: D16193425

fbshipit-source-id: 8ee3a0dc46768d7cb572bf383be1ce2b450c44c9
---
 src/PackWeightsForConv.cc | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'src')

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");
-- 
cgit v1.2.3