|
|
|
@ -125,9 +125,11 @@ TEST_F(CommonTest, sgdUpdate) {
|
|
|
|
|
const size_t alignHeader[] = {0, 2, 3, 5, 7, 8};
|
|
|
|
|
for (auto& size : sizeVec_) {
|
|
|
|
|
real *gradientBuffer, *valueBuffer, *momentumBuffer;
|
|
|
|
|
posix_memalign((void**)&gradientBuffer, 32, sizeof(real) * size);
|
|
|
|
|
posix_memalign((void**)&valueBuffer, 32, sizeof(real) * size);
|
|
|
|
|
posix_memalign((void**)&momentumBuffer, 32, sizeof(real) * size);
|
|
|
|
|
CHECK_EQ(posix_memalign((void**)&gradientBuffer, 32, sizeof(real) * size),
|
|
|
|
|
0);
|
|
|
|
|
CHECK_EQ(posix_memalign((void**)&valueBuffer, 32, sizeof(real) * size), 0);
|
|
|
|
|
CHECK_EQ(posix_memalign((void**)&momentumBuffer, 32, sizeof(real) * size),
|
|
|
|
|
0);
|
|
|
|
|
|
|
|
|
|
for (size_t i = 0; i < size; i++) {
|
|
|
|
|
gradientBuffer[i] = 1.0;
|
|
|
|
|