cuberl_types.h

Go to the documentation of this file.

#ifndef CUBEAI_TYPES_H
#define CUBEAI_TYPES_H
 
#include "bitrl/bitrl_types.h"
#include "cuberl/base/cubeai_config.h"
 
#ifdef USE_PYTORCH
#include <torch/torch.h>
#endif
 
#include <cstddef>
#include <string>
 
 
namespace cuberl
{
    
    using namespace bitrl;
    
    using bitrl::int_t;
    
    using bitrl::lint_t;
 
    using bitrl::float_t;
    
    using bitrl::real_t;
    
    using bitrl::DynMat;
 
    using bitrl::SquareMat;
 
    using bitrl::Mat;
 
    using bitrl::DynVec;
 
    using bitrl::FloatVec;
    
    using bitrl::RealVec;
 
    using bitrl::Null;
    
    using bitrl::DeviceType;
    
#ifdef USE_PYTORCH
   struct torch_t {};
 
   typedef long int torch_int_t;
 
   typedef torch::Tensor torch_tensor_t;
   
   
    template<typename T>
    struct torch_tensor_value_type{
        typedef T value_type;
    };
#endif
 
    
 
}
 
#endif