From 97df75b2c2e284fb06ce9fc9ed613b12b1c1f617 Mon Sep 17 00:00:00 2001 From: Liu Yiqun Date: Mon, 30 Dec 2019 05:27:37 +0000 Subject: [PATCH] Add comments for CHECK of size. --- lite/core/tensor.h | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/lite/core/tensor.h b/lite/core/tensor.h index b7dd57d9b5..92801066df 100644 --- a/lite/core/tensor.h +++ b/lite/core/tensor.h @@ -38,7 +38,7 @@ class TensorLite; using DDim = lite::DDimLite; using Tensor = lite::TensorLite; -template +template class DimVector { public: DimVector() { @@ -48,7 +48,9 @@ class DimVector { size_t size() const { return size_; } void resize(size_t new_size) { - CHECK_LE(new_size, DimLength); + CHECK_LE(new_size, DimLength) + << "Expected the number of dimentations <= " << DimLength + << ", received " << new_size << "."; size_ = new_size; } @@ -63,10 +65,10 @@ class DimVector { size_t size_{0}; }; -constexpr int kMaxDimLength = 10; - class DDimLite { public: + constexpr static size_t kMaxDimLength = 10; + using value_type = int64_t; using DDimVector = DimVector; -- GitLab