mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
085e2f7bdd
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/65610 - Replace HIP_PLATFORM_HCC with USE_ROCM - Dont rely on CUDA_VERSION or HIP_VERSION and use USE_ROCM and ROCM_VERSION. - In the next PR - Will be removing the mapping from CUDA_VERSION to HIP_VERSION and CUDA to HIP in hipify. - HIP_PLATFORM_HCC is deprecated, so will add HIP_PLATFORM_AMD to support HIP host code compilation on gcc. cc jeffdaily sunway513 jithunnair-amd ROCmSupport amathews-amd Reviewed By: jbschlosser Differential Revision: D30909053 Pulled By: ezyang fbshipit-source-id: 224a966ebf1aaec79beccbbd686fdf3d49267e06
87 lines
3.2 KiB
C
87 lines
3.2 KiB
C
// Automatically generated header file for caffe2 macros. These
|
|
// macros are used to build the Caffe2 binary, and if you are
|
|
// building a dependent library, they will need to be set as well
|
|
// for your program to link correctly.
|
|
|
|
#pragma once
|
|
|
|
// Caffe2 version. The plan is to increment the minor version every other week
|
|
// as a track point for bugs, until we find a proper versioning cycle.
|
|
|
|
#define CAFFE2_VERSION_MAJOR @CAFFE2_VERSION_MAJOR@
|
|
#define CAFFE2_VERSION_MINOR @CAFFE2_VERSION_MINOR@
|
|
#define CAFFE2_VERSION_PATCH @CAFFE2_VERSION_PATCH@
|
|
|
|
static_assert(
|
|
CAFFE2_VERSION_MINOR < 100,
|
|
"Programming error: you set a minor version that is too big.");
|
|
static_assert(
|
|
CAFFE2_VERSION_PATCH < 100,
|
|
"Programming error: you set a patch version that is too big.");
|
|
|
|
#define CAFFE2_VERSION \
|
|
(CAFFE2_VERSION_MAJOR * 10000 + CAFFE2_VERSION_MINOR * 100 + \
|
|
CAFFE2_VERSION_PATCH)
|
|
|
|
#cmakedefine CAFFE2_BUILD_SHARED_LIBS
|
|
#cmakedefine CAFFE2_FORCE_FALLBACK_CUDA_MPI
|
|
#cmakedefine CAFFE2_HAS_MKL_DNN
|
|
#cmakedefine CAFFE2_HAS_MKL_SGEMM_PACK
|
|
#cmakedefine CAFFE2_PERF_WITH_AVX
|
|
#cmakedefine CAFFE2_PERF_WITH_AVX2
|
|
#cmakedefine CAFFE2_PERF_WITH_AVX512
|
|
#cmakedefine CAFFE2_THREADPOOL_MAIN_IMBALANCE
|
|
#cmakedefine CAFFE2_THREADPOOL_STATS
|
|
#cmakedefine CAFFE2_USE_EXCEPTION_PTR
|
|
#cmakedefine CAFFE2_USE_ACCELERATE
|
|
#cmakedefine CAFFE2_USE_CUDNN
|
|
#cmakedefine CAFFE2_USE_EIGEN_FOR_BLAS
|
|
#cmakedefine CAFFE2_USE_FBCODE
|
|
#cmakedefine CAFFE2_USE_GOOGLE_GLOG
|
|
#cmakedefine CAFFE2_USE_LITE_PROTO
|
|
#cmakedefine CAFFE2_USE_MKL
|
|
#cmakedefine CAFFE2_USE_MKLDNN
|
|
#cmakedefine CAFFE2_USE_NVTX
|
|
#cmakedefine CAFFE2_USE_TRT
|
|
|
|
#ifndef EIGEN_MPL2_ONLY
|
|
#cmakedefine EIGEN_MPL2_ONLY
|
|
#endif
|
|
|
|
// Useful build settings that are recorded in the compiled binary
|
|
#define CAFFE2_BUILD_STRINGS { \
|
|
{"TORCH_VERSION", "${TORCH_VERSION}"}, \
|
|
{"CXX_COMPILER", "${CMAKE_CXX_COMPILER}"}, \
|
|
{"CXX_FLAGS", "${CMAKE_CXX_FLAGS}"}, \
|
|
{"BUILD_TYPE", "${CMAKE_BUILD_TYPE}"}, \
|
|
{"BLAS_INFO", "${BLAS_INFO}"}, \
|
|
{"LAPACK_INFO", "${LAPACK_INFO}"}, \
|
|
{"USE_CUDA", "${USE_CUDA}"}, \
|
|
{"USE_ROCM", "${USE_ROCM}"}, \
|
|
{"CUDA_VERSION", "${CUDA_VERSION}"}, \
|
|
{"ROCM_VERSION", "${ROCM_VERSION}"}, \
|
|
{"USE_CUDNN", "${USE_CUDNN}"}, \
|
|
{"CUDNN_VERSION", "${CUDNN_VERSION}"}, \
|
|
{"USE_NCCL", "${USE_NCCL}"}, \
|
|
{"USE_MPI", "${USE_MPI}"}, \
|
|
{"USE_GFLAGS", "${USE_GFLAGS}"}, \
|
|
{"USE_GLOG", "${USE_GLOG}"}, \
|
|
{"USE_GLOO", "${USE_GLOI}"}, \
|
|
{"USE_NNPACK", "${USE_NNPACK}"}, \
|
|
{"USE_OPENMP", "${USE_OPENMP}"}, \
|
|
{"FORCE_FALLBACK_CUDA_MPI", "${CAFFE2_FORCE_FALLBACK_CUDA_MPI}"}, \
|
|
{"HAS_MKL_DNN", "${CAFFE2_HAS_MKL_DNN}"}, \
|
|
{"HAS_MKL_SGEMM_PACK", "${CAFFE2_HAS_MKL_SGEMM_PACK}"}, \
|
|
{"PERF_WITH_AVX", "${CAFFE2_PERF_WITH_AVX}"}, \
|
|
{"PERF_WITH_AVX2", "${CAFFE2_PERF_WITH_AVX2}"}, \
|
|
{"PERF_WITH_AVX512", "${CAFFE2_PERF_WITH_AVX512}"}, \
|
|
{"USE_EXCEPTION_PTR", "${CAFFE2_USE_EXCEPTION_PTR}"}, \
|
|
{"USE_ACCELERATE", "${CAFFE2_USE_ACCELERATE}"}, \
|
|
{"USE_EIGEN_FOR_BLAS", "${CAFFE2_USE_EIGEN_FOR_BLAS}"}, \
|
|
{"USE_LITE_PROTO", "${CAFFE2_USE_LITE_PROTO}"}, \
|
|
{"USE_MKL", "${CAFFE2_USE_MKL}"}, \
|
|
{"USE_MKLDNN", "${CAFFE2_USE_MKLDNN}"}, \
|
|
{"USE_NVTX", "${CAFFE2_USE_NVTX}"}, \
|
|
{"USE_TRT", "${CAFFE2_USE_TRT}"}, \
|
|
}
|