Skip to content

Commit 0fca168

Browse files
committed
temp
1 parent e9be336 commit 0fca168

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

paddle/fluid/operators/math/matrix_bit_code.cc

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -102,6 +102,8 @@ void MatrixBitCodeFunctor<T>::MulGradWeight(const framework::LoDTensor& tmat,
102102
size_t input_width = input.dims()[1];
103103
size_t tmat_width = tmat.dims()[1];
104104
size_t weight_width = weight->dims()[1];
105+
VLOG(30) << "sparse w_grad dims is [" << weight->dims()[0] << " ,"
106+
<< weight->dims()[1] << " ]";
105107
auto tmat_value = tmat.data<T>();
106108
auto weight_value = weight->data<T>();
107109
auto input_value = input.data<T>();
@@ -127,6 +129,8 @@ void MatrixBitCodeFunctor<T>::MulGradWeight(const framework::LoDTensor& tmat,
127129
size_t input_width = input.dims()[1];
128130
size_t tmat_width = tmat.dims()[1];
129131
size_t weight_width = weight->value().dims()[1];
132+
VLOG(30) << "sparse w_grad dims is: [" << weight->value().dims()[0] << " ,"
133+
<< weight->value().dims()[1] << " ]";
130134
auto tmat_value = tmat.data<T>();
131135
auto weight_value = weight->mutable_value()->data<T>();
132136
auto input_value = input.data<T>();

0 commit comments

Comments
 (0)