mirror of
https://gitlab.com/libeigen/eigen.git
synced 2025-07-29 00:02:40 +08:00
7d08fa805a
* Abstracting the pointer type so that both SYCL memory and pointer can be captured. * Converting SYCL virtual pointer to SYCL device memory in Eigen evaluator class. * Binding SYCL placeholder accessor to command group handler by using bind method in Eigen evaluator node. * Adding SYCL macro for controlling loop unrolling. * Modifying the TensorDeviceSycl.h and SYCL executor method to adopt the above changes.
98 lines
2.8 KiB
C++
98 lines
2.8 KiB
C++
// This file is part of Eigen, a lightweight C++ template library
|
|
// for linear algebra.
|
|
//
|
|
// Copyright (C) 2015 Benoit Steiner <benoit.steiner.goog@gmail.com>
|
|
//
|
|
// This Source Code Form is subject to the terms of the Mozilla
|
|
// Public License v. 2.0. If a copy of the MPL was not distributed
|
|
// with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
#ifndef EIGEN_CXX11_TENSOR_TENSOR_META_MACROS_H
|
|
#define EIGEN_CXX11_TENSOR_TENSOR_META_MACROS_H
|
|
|
|
|
|
/** use this macro in sfinae selection in templated functions
|
|
*
|
|
* template<typename T,
|
|
* typename std::enable_if< isBanana<T>::value , int >::type = 0
|
|
* >
|
|
* void foo(){}
|
|
*
|
|
* becomes =>
|
|
*
|
|
* template<typename TopoType,
|
|
* SFINAE_ENABLE_IF( isBanana<T>::value )
|
|
* >
|
|
* void foo(){}
|
|
*/
|
|
|
|
// SFINAE requires variadic templates
|
|
#if !defined(EIGEN_GPUCC)
|
|
#if EIGEN_HAS_VARIADIC_TEMPLATES
|
|
// SFINAE doesn't work for gcc <= 4.7
|
|
#ifdef EIGEN_COMP_GNUC
|
|
#if EIGEN_GNUC_AT_LEAST(4,8)
|
|
#define EIGEN_HAS_SFINAE
|
|
#endif
|
|
#else
|
|
#define EIGEN_HAS_SFINAE
|
|
#endif
|
|
#endif
|
|
#endif
|
|
|
|
#define EIGEN_SFINAE_ENABLE_IF( __condition__ ) \
|
|
typename internal::enable_if< ( __condition__ ) , int >::type = 0
|
|
|
|
|
|
#if EIGEN_HAS_CONSTEXPR
|
|
#define EIGEN_CONSTEXPR constexpr
|
|
#else
|
|
#define EIGEN_CONSTEXPR
|
|
#endif
|
|
|
|
|
|
#if EIGEN_OS_WIN || EIGEN_OS_WIN64
|
|
#define EIGEN_SLEEP(n) Sleep(n)
|
|
#elif EIGEN_OS_GNULINUX
|
|
#define EIGEN_SLEEP(n) usleep(n * 1000);
|
|
#else
|
|
#define EIGEN_SLEEP(n) sleep(std::max<unsigned>(1, n/1000))
|
|
#endif
|
|
|
|
// Define a macro to use a reference on the host but a value on the device
|
|
#if defined(SYCL_DEVICE_ONLY)
|
|
#define EIGEN_DEVICE_REF
|
|
#else
|
|
#define EIGEN_DEVICE_REF &
|
|
#endif
|
|
|
|
// Define a macro for catching SYCL exceptions if exceptions are enabled
|
|
#if defined(EIGEN_EXCEPTIONS)
|
|
#define EIGEN_SYCL_TRY_CATCH(X) \
|
|
do { \
|
|
try { X; } \
|
|
catch(const cl::sycl::exception& e) { \
|
|
std::cerr << "SYCL exception at " \
|
|
<< __FILE__ << ":" << __LINE__ << std::endl \
|
|
<< e.what() << std::endl; \
|
|
std::rethrow_exception(std::current_exception()); \
|
|
} \
|
|
} while (false)
|
|
#else
|
|
#define EIGEN_SYCL_TRY_CATCH(X) X
|
|
#endif
|
|
|
|
// Define a macro if local memory flags are unset or one of them is set
|
|
// Setting both flags is the same as unsetting them
|
|
#if (!defined(EIGEN_SYCL_LOCAL_MEM) && !defined(EIGEN_SYCL_NO_LOCAL_MEM)) || \
|
|
(defined(EIGEN_SYCL_LOCAL_MEM) && defined(EIGEN_SYCL_NO_LOCAL_MEM))
|
|
#define EIGEN_SYCL_LOCAL_MEM_UNSET_OR_ON 1
|
|
#define EIGEN_SYCL_LOCAL_MEM_UNSET_OR_OFF 1
|
|
#elif defined(EIGEN_SYCL_LOCAL_MEM) && !defined(EIGEN_SYCL_NO_LOCAL_MEM)
|
|
#define EIGEN_SYCL_LOCAL_MEM_UNSET_OR_ON 1
|
|
#elif !defined(EIGEN_SYCL_LOCAL_MEM) && defined(EIGEN_SYCL_NO_LOCAL_MEM)
|
|
#define EIGEN_SYCL_LOCAL_MEM_UNSET_OR_OFF 1
|
|
#endif
|
|
|
|
#endif
|