DeviceTypeConfig.cu.h

Go to the documentation of this file.

// ---------------------------------------------------------------------
//
// Copyright (c) 2017-2025 The Regents of the University of Michigan and DFT-FE
// authors.
//
// This file is part of the DFT-FE code.
//
// The DFT-FE code is free software; you can use it, redistribute
// it, and/or modify it under the terms of the GNU Lesser General
// Public License as published by the Free Software Foundation; either
// version 2.1 of the License, or (at your option) any later version.
// The full text of the license can be found in the file LICENSE at
// the top level of the DFT-FE distribution.
//
// ---------------------------------------------------------------------
//
#ifndef dftfeDeviceTypeConfig_cuh
#define dftfeDeviceTypeConfig_cuh
 
#include <cuComplex.h>
#include <cublas_v2.h>
namespace dftfe
{
  namespace utils
  {
    typedef cudaError_t     deviceError_t;
    typedef cudaStream_t    deviceStream_t;
    typedef cudaEvent_t     deviceEvent_t;
    typedef cuDoubleComplex deviceDoubleComplex;
    typedef cuFloatComplex  deviceFloatComplex;
 
    // static consts
    static const deviceError_t deviceSuccess = cudaSuccess;
 
    // vendor blas related typedef and static consts
    typedef cublasHandle_t      deviceBlasHandle_t;
    typedef cublasOperation_t   deviceBlasOperation_t;
    typedef cublasStatus_t      deviceBlasStatus_t;
    typedef cublasComputeType_t deviceBlasComputeType_t;
    typedef cudaDataType        deviceDataType_t;
 
    static const cublasOperation_t   DEVICEBLAS_OP_N = CUBLAS_OP_N;
    static const cublasOperation_t   DEVICEBLAS_OP_T = CUBLAS_OP_T;
    static const cublasOperation_t   DEVICEBLAS_OP_C = CUBLAS_OP_C;
    static const cublasComputeType_t DEVICEBLAS_COMPUTE_32F =
      CUBLAS_COMPUTE_32F;
    static const cublasComputeType_t DEVICEBLAS_COMPUTE_32F_FAST_TF32 =
      CUBLAS_COMPUTE_32F_FAST_TF32;
    static const cublasComputeType_t DEVICEBLAS_COMPUTE_32F_FAST_16BF =
      CUBLAS_COMPUTE_32F_FAST_16BF;
    static const cublasComputeType_t DEVICEBLAS_COMPUTE_32F_FAST_16F =
      CUBLAS_COMPUTE_32F_FAST_16F;
    static const cublasGemmAlgo_t DEVICEBLAS_GEMM_DEFAULT = CUBLAS_GEMM_DEFAULT;
    static const cudaDataType     DEVICE_R_64F            = CUDA_R_64F;
    static const cudaDataType     DEVICE_R_32F            = CUDA_R_32F;
    static const cudaDataType     DEVICE_R_16F            = CUDA_R_16F;
    static const cudaDataType     DEVICE_R_16BF           = CUDA_R_16BF;
    static const cudaDataType     DEVICE_C_64F            = CUDA_C_64F;
    static const cudaDataType     DEVICE_C_32F            = CUDA_C_32F;
    static const cudaDataType     DEVICE_C_16F            = CUDA_C_16F;
    static const cudaDataType     DEVICE_C_16BF           = CUDA_C_16BF;
 
  } // namespace utils
} // namespace dftfe
 
#endif // dftfeDeviceTypeConfig_cuh