Brutex
/
cavis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

two tests fixes 

Signed-off-by: raver119 <raver119@gmail.com>
master
raver119 2020-01-24 15:26:46 +03:00
parent 5d69069177
commit 99cfb88933
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
libnd4j/tests_cpu/layers_tests/DataBufferTests.cpp
View File

@ -48,7 +48,7 @@ TEST_F(DataBufferTests, test_alloc_limit_1) {
auto odUse = MemoryCounter::getInstance()->allocatedDevice(deviceId); auto odUse = MemoryCounter::getInstance()->allocatedDevice(deviceId);
auto ogUse = MemoryCounter::getInstance()->allocatedGroup(MemoryType::HOST); auto ogUse = MemoryCounter::getInstance()->allocatedGroup(MemoryType::HOST);
auto limitSize = 150 * 1024 * 1024; auto limitSize = odUse + (150 * 1024 * 1024);
auto allocSize = 100000000; auto allocSize = 100000000;
MemoryCounter::getInstance()->setDeviceLimit(deviceId, odLimit + limitSize); MemoryCounter::getInstance()->setDeviceLimit(deviceId, odLimit + limitSize);

4
libnd4j/tests_cpu/layers_tests/DataBufferTestsCuda.cu
View File

@ -51,7 +51,7 @@ TEST_F(DataBufferTestsCuda, test_alloc_limit_1) {
auto opUse = MemoryCounter::getInstance()->allocatedGroup(MemoryType::HOST); auto opUse = MemoryCounter::getInstance()->allocatedGroup(MemoryType::HOST);
auto osUse = MemoryCounter::getInstance()->allocatedGroup(MemoryType::DEVICE); auto osUse = MemoryCounter::getInstance()->allocatedGroup(MemoryType::DEVICE);
auto limitSize = 150000000; auto limitSize = odUse + 150000000;
auto allocSize = 100000000; auto allocSize = 100000000;
MemoryCounter::getInstance()->setDeviceLimit(deviceId, odLimit + limitSize); MemoryCounter::getInstance()->setDeviceLimit(deviceId, odLimit + limitSize);
@ -72,7 +72,7 @@ TEST_F(DataBufferTestsCuda, test_alloc_limit_1) {
// this allocation should fail, since we're allocating too much // this allocation should fail, since we're allocating too much
try { try {
DataBuffer bufferFailed(allocSize, DataType::INT32); DataBuffer bufferFailed(allocSize + 1, DataType::INT32);
ASSERT_TRUE(false); ASSERT_TRUE(false);
} catch (allocation_exception &e) { } catch (allocation_exception &e) {
// we expect exception here // we expect exception here