Line data    Source code

       1             : #pragma once
       2             : 
       3             : #include "CUDADefines.h"
       4             : 
       5             : namespace elsa::fn
       6             : {
       7             :     namespace detail
       8             :     {
       9             :         struct SquareFn {
      10             :             template <class T>
      11             :             __host__ __device__ constexpr T operator()(const T& arg) const noexcept
      12      353197 :             {
      13      353197 :                 return arg * arg;
      14      353197 :             }
      15             :         };
      16             :     } // namespace detail
      17             : 
      18             :     static constexpr __device__ detail::SquareFn square;
      19             : } // namespace elsa::fn