data.hpp

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: Copyright (c) 2019-2025, NVIDIA CORPORATION.
 * SPDX-License-Identifier: Apache-2.0
 */
 
#pragma once
 
namespace MLCommon {
namespace Matrix {
 
template <typename Type>
struct Data {
  Data() : ptr(nullptr), totalSize(0) {}
  Data(Type* _ptr, size_t _n_elements) : ptr(_ptr), totalSize(_n_elements * sizeof(Type)) {}
 
  Type* ptr = nullptr;
 
  size_t totalSize = (size_t)0;
 
  size_t numElements() const { return totalSize / sizeof(Type); }
};
 
typedef Data<float> floatData_t;
typedef Data<double> doubleData_t;
 
};  // end namespace Matrix
};  // end namespace MLCommon