some bug fix

avx_docs
hedaoyuan 8 years ago
parent 956b661cd9
commit 11479901a7

@ -93,6 +93,11 @@ float construct(int height, int width) {
return 0.5; return 0.5;
} }
template <>
double construct(int height, int width) {
return 0.5;
}
template <> template <>
size_t construct(int height, int width) { size_t construct(int height, int width) {
size_t offset = std::rand() % (height < width ? height : width); size_t offset = std::rand() % (height < width ? height : width);

@ -290,8 +290,8 @@ void testMatrixDeepSwap(int height, int width) {
// swap matrix cpuA and cpuB // swap matrix cpuA and cpuB
cpuA->deepSwap(*cpuB); cpuA->deepSwap(*cpuB);
MatrixCheckEqual(*cpuA, *cpuCopyB); TensorCheckEqual(*cpuA, *cpuCopyB);
MatrixCheckEqual(*cpuB, *cpuCopyA); TensorCheckEqual(*cpuB, *cpuCopyA);
} }
void testMatrixTranspose(int height, int width) { void testMatrixTranspose(int height, int width) {

Loading…
Cancel
Save