parent
fbf7c9d38b
commit
ac0d249f07
|
@ -101,7 +101,7 @@ ELSE()
|
||||||
endif()
|
endif()
|
||||||
ENDIF()
|
ENDIF()
|
||||||
|
|
||||||
if ("${CMAKE_CXX_COMPILER_ID}" STREQUAL "Clang")
|
if ("${CMAKE_CXX_COMPILER_ID}" STREQUAL "Clang" OR "${CMAKE_CXX_COMPILER_ID}" STREQUAL "AppleClang")
|
||||||
# using Clang
|
# using Clang
|
||||||
SET( CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${ARCH_TUNE}")
|
SET( CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${ARCH_TUNE}")
|
||||||
|
|
||||||
|
|
|
@ -3107,7 +3107,6 @@ const char* lastErrorMessage() {
|
||||||
int binaryLevel() {
|
int binaryLevel() {
|
||||||
#ifdef CPU_FEATURES
|
#ifdef CPU_FEATURES
|
||||||
|
|
||||||
|
|
||||||
#if defined(F_X64)
|
#if defined(F_X64)
|
||||||
return 1;
|
return 1;
|
||||||
#elif defined (F_AVX2)
|
#elif defined (F_AVX2)
|
||||||
|
|
|
@ -60,6 +60,10 @@ public:
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
|
TEST_F(PlaygroundTests, test_avx) {
|
||||||
|
nd4j_printf("Optimal level: %i; Binary level: %i;\n", ::optimalLevel(), ::binaryLevel());
|
||||||
|
}
|
||||||
|
|
||||||
/*
|
/*
|
||||||
TEST_F(PlaygroundTests, test_s_0) {
|
TEST_F(PlaygroundTests, test_s_0) {
|
||||||
auto x = NDArrayFactory::create<float>('c', {32, 112, 112, 16});
|
auto x = NDArrayFactory::create<float>('c', {32, 112, 112, 16});
|
||||||
|
|
Loading…
Reference in New Issue