Commit 8e8c18ac authored by Robert Kimball's avatar Robert Kimball Committed by Scott Cyphers

Call ngraph_malloc instead of malloc so it throws if out of memory (#3681)

parent b408bbe6
...@@ -44,7 +44,7 @@ runtime::AlignedBuffer::AlignedBuffer(size_t byte_size, size_t alignment, Alloca ...@@ -44,7 +44,7 @@ runtime::AlignedBuffer::AlignedBuffer(size_t byte_size, size_t alignment, Alloca
} }
else else
{ {
m_allocated_buffer = static_cast<char*>(malloc(allocation_size)); m_allocated_buffer = static_cast<char*>(ngraph_malloc(allocation_size));
} }
m_aligned_buffer = m_allocated_buffer; m_aligned_buffer = m_allocated_buffer;
size_t mod = size_t(m_aligned_buffer) % alignment; size_t mod = size_t(m_aligned_buffer) % alignment;
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment