Lines Matching +full:cmake +full:- +full:core
1 # This ill-named file does a number of things:
2 # - Installs Caffe2 header files (this has nothing to do with code generation)
3 # - Configures caffe2/core/macros.h
4 # - Creates an ATen target for its generated C++ files and adds it
6 # - Reads build lists defined in build_variables.bzl
38 # ---[ Write the macros file
40 ${CMAKE_CURRENT_LIST_DIR}/../caffe2/core/macros.h.in
41 ${CMAKE_BINARY_DIR}/caffe2/core/macros.h)
43 # ---[ Installing the header files
48 install(DIRECTORY ${CMAKE_CURRENT_LIST_DIR}/../aten/src/ATen/core
52 install(FILES ${CMAKE_BINARY_DIR}/caffe2/core/macros.h
53 DESTINATION include/caffe2/core)
55 # ---[ ATen specific
60 set(OPT_FLAG "-O3 ")
67 …${CMAKE_CURRENT_LIST_DIR}/../aten/src/ATen/MapAllocator.cpp PROPERTIES COMPILE_FLAGS "-fno-openmp")
83 PROPERTIES COMPILE_FLAGS "-gencode arch=compute_90a,code=sm_90a")
89 set(GEN_ROCM_FLAG --rocm)
94 set(GEN_MPS_FLAG --mps)
100 list(APPEND CUSTOM_BUILD_FLAGS --backend_whitelist CPU QuantizedCPU Vulkan)
102 list(APPEND CUSTOM_BUILD_FLAGS --backend_whitelist CPU QuantizedCPU)
108 … message(STATUS "Running tracing-based selective build given operator list: ${SELECTED_OP_LIST}")
110 --op_selection_yaml_path ${SELECTED_OP_LIST})
113 "You have to run tracing-based selective build with dynamic dispatch.\n"
124 --static_dispatch_backend ${STATIC_DISPATCH_BACKEND})
130 list(APPEND CUSTOM_BUILD_FLAGS --skip_dispatcher_op_registration)
132 "${Python_EXECUTABLE}" -m tools.jit.gen_unboxing
133 --source-path ${CMAKE_CURRENT_LIST_DIR}/../aten/src/ATen
134 --install_dir ${CMAKE_BINARY_DIR}/aten/src/ATen
138 --TEST_ONLY_op_registration_allowlist_yaml_path "${SELECTED_OP_LIST}")
142 --output-dependencies ${CMAKE_BINARY_DIR}/aten/src/ATen/generated_unboxing_sources.cmake
146 COMMAND ${GEN_UNBOXING_COMMAND_sources} --dry-run
154 include("${CMAKE_BINARY_DIR}/aten/src/ATen/generated_unboxing_sources.cmake")
159 ${CMAKE_BINARY_DIR}/aten/src/ATen/generated_unboxing_sources.cmake
172 list(APPEND GEN_PER_OPERATOR_FLAG "--per-operator-headers")
176 "${Python_EXECUTABLE}" -m torchgen.gen
177 --source-path ${CMAKE_CURRENT_LIST_DIR}/../aten/src/ATen
178 --install_dir ${CMAKE_BINARY_DIR}/aten/src/ATen
193 # We work around this by generating a .cmake file which is
195 # ever changes then cmake will be re-run automatically because it
200 --generate ${gen_type}
201 --output-dependencies ${CMAKE_BINARY_DIR}/aten/src/ATen/generated_${gen_type}.cmake
206 COMMAND ${GEN_COMMAND_${gen_type}} --dry-run
215 include("${CMAKE_BINARY_DIR}/aten/src/ATen/generated_${gen_type}.cmake")
216 include("${CMAKE_BINARY_DIR}/aten/src/ATen/core_generated_${gen_type}.cmake")
217 include("${CMAKE_BINARY_DIR}/aten/src/ATen/cpu_vec_generated_${gen_type}.cmake")
218 include("${CMAKE_BINARY_DIR}/aten/src/ATen/cuda_generated_${gen_type}.cmake")
219 include("${CMAKE_BINARY_DIR}/aten/src/ATen/ops_generated_${gen_type}.cmake")
231 ${CMAKE_BINARY_DIR}/aten/src/ATen/generated_${gen_type}.cmake
232 ${CMAKE_BINARY_DIR}/aten/src/ATen/ops_generated_${gen_type}.cmake
233 ${CMAKE_BINARY_DIR}/aten/src/ATen/core_generated_${gen_type}.cmake
234 ${CMAKE_BINARY_DIR}/aten/src/ATen/cpu_vec_generated_${gen_type}.cmake
235 ${CMAKE_BINARY_DIR}/aten/src/ATen/cuda_generated_${gen_type}.cmake
245 # not tracked correctly in CMake. We make the libATen.so depend explicitly
271 set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -DHAVE_AVX512_CPU_DEFINITION")
276 … list(APPEND CPU_CAPABILITY_FLAGS "${OPT_FLAG} -mavx512f -mavx512bw -mavx512vl -mavx512dq -mfma")
281 set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -DHAVE_AVX2_CPU_DEFINITION")
286 …https://stackoverflow.com/questions/52626726/why-doesnt-gcc-resolve-mm256-loadu-pd-as-single-vmovu…
287 …check_cxx_compiler_flag("-mno-avx256-split-unaligned-load -mno-avx256-split-unaligned-store" COMPI…
289 …set(CPU_NO_AVX256_SPLIT_FLAGS "-mno-avx256-split-unaligned-load -mno-avx256-split-unaligned-store")
296 message("-- ATen AVX2 kernels will use 32 ymm registers")
300 … list(APPEND CPU_CAPABILITY_FLAGS "${OPT_FLAG} -march=native ${CPU_NO_AVX256_SPLIT_FLAGS}")
308 … list(APPEND CPU_CAPABILITY_FLAGS "${OPT_FLAG} -mavx2 -mfma -mf16c ${CPU_NO_AVX256_SPLIT_FLAGS}")
314 SET(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -DHAVE_VSX_CPU_DEFINITION")
320 SET(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -DHAVE_ZVECTOR_CPU_DEFINITION")
326 math(EXPR NUM_CPU_CAPABILITY_NAMES "${NUM_CPU_CAPABILITY_NAMES}-1")
329 # See NOTE [ Linking AVX and non-AVX files ]
334 configure_file("${PROJECT_SOURCE_DIR}/cmake/IncludeSource.cpp.in" ${NEW_IMPL})
340 set(EXTRA_FLAGS "-DCPU_CAPABILITY=${CPU_CAPABILITY} -DCPU_CAPABILITY_${CPU_CAPABILITY}")
342 # Disable certain warnings for GCC-9.X
346 set(EXTRA_FLAGS "${EXTRA_FLAGS} -Wno-uninitialized")
350 set(EXTRA_FLAGS "${EXTRA_FLAGS} -Wno-deprecated-copy")
374 COMMAND "${Python_EXECUTABLE}" -c