Program Listing for File type_util.hpp
↰ Return to documentation for file (morpheus/_lib/include/morpheus/utilities/type_util.hpp
)
#pragma once
#include "morpheus/utilities/type_util_detail.hpp"
#include <cudf/types.hpp>
#include <cudf/utilities/traits.hpp>
#include <memory>
#include <stdexcept>
namespace morpheus {
/****** Component public implementations *******************/
/****** DType****************************************/
struct DType : DataType // NOLINT
{
DType(const DataType& dtype);
DType(TypeId tid = TypeId::EMPTY);
// Cudf representation
cudf::type_id cudf_type_id() const;
// Returns the triton string representation
std::string triton_str() const;
// from template
template <typename T>
static DType create()
{
return DType(DataType::create<T>());
}
// From cudf
static DType from_cudf(cudf::type_id tid);
// From triton
static DType from_triton(const std::string& type_str);
};
// end of group
} // namespace morpheus