parent
fbf7c9d38b
commit
ac0d249f07
|
@ -101,7 +101,7 @@ ELSE()
|
|||
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
|
||||
SET( CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${ARCH_TUNE}")
|
||||
|
||||
|
|
|
@ -3107,7 +3107,6 @@ const char* lastErrorMessage() {
|
|||
int binaryLevel() {
|
||||
#ifdef CPU_FEATURES
|
||||
|
||||
|
||||
#if defined(F_X64)
|
||||
return 1;
|
||||
#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) {
|
||||
auto x = NDArrayFactory::create<float>('c', {32, 112, 112, 16});
|
||||
|
|
Loading…
Reference in New Issue