parent
5de00682ce
commit
ffae024cda
|
@ -23,7 +23,7 @@
|
||||||
#include <exceptions/cuda_exception.h>
|
#include <exceptions/cuda_exception.h>
|
||||||
#include <thread>
|
#include <thread>
|
||||||
|
|
||||||
#ifdef IOS_BUILD
|
#if defined(IOS_BUILD) || defined(APPLE_BUILD) || defined(ANDROID_BUILD)
|
||||||
nd4j::ContextBuffers contextBuffers = nd4j::ContextBuffers();
|
nd4j::ContextBuffers contextBuffers = nd4j::ContextBuffers();
|
||||||
#else
|
#else
|
||||||
thread_local nd4j::ContextBuffers contextBuffers = nd4j::ContextBuffers();
|
thread_local nd4j::ContextBuffers contextBuffers = nd4j::ContextBuffers();
|
||||||
|
|
Loading…
Reference in New Issue