Skip to content

Commit f5e8b8c

Browse files
lcxywferajeevsrao
authored andcommitted
fix normalize plugin deserialization
Signed-off-by: Yao He <lcxywfe@gmail.com>
1 parent 8cca9a8 commit f5e8b8c

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

plugin/normalizePlugin/normalizePlugin.cpp

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -71,8 +71,9 @@ Normalize::Normalize(const void* buffer, size_t length)
7171
channelShared = read<bool>(d);
7272
eps = read<float>(d);
7373

74-
int nbWeights = read<int>(d);
75-
mWeights = deserializeToDevice(d, nbWeights);
74+
mNbWeights = read<int>(d);
75+
mWeights = deserializeToDevice(d, mNbWeights);
76+
cublasCreate(&mCublas);
7677
ASSERT(d == a + length);
7778
}
7879

0 commit comments

Comments
 (0)