mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-08 07:39:33 +01:00
0478d32cb8
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/13916 Reviewed By: smessmer Differential Revision: D13046722 fbshipit-source-id: 1583d3170d60e22f0a535cd1fd56bdf928186f5d
53 lines
1.7 KiB
C++
53 lines
1.7 KiB
C++
//===- llvm/ADT/SmallVector.cpp - 'Normally small' vectors ----------------===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file implements the SmallVector class.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
// ATen: modified from llvm::SmallVector.
|
|
// replaced report_bad_alloc_error with std::bad_alloc
|
|
|
|
#include <c10/util/SmallVector.h>
|
|
|
|
namespace c10 {
|
|
|
|
/// grow_pod - This is an implementation of the grow() method which only works
|
|
/// on POD-like datatypes and is out of line to reduce code duplication.
|
|
void SmallVectorBase::grow_pod(
|
|
void* FirstEl,
|
|
size_t MinSizeInBytes,
|
|
size_t TSize) {
|
|
size_t CurSizeBytes = size_in_bytes();
|
|
size_t NewCapacityInBytes = 2 * capacity_in_bytes() + TSize; // Always grow.
|
|
if (NewCapacityInBytes < MinSizeInBytes)
|
|
NewCapacityInBytes = MinSizeInBytes;
|
|
|
|
void* NewElts;
|
|
if (BeginX == FirstEl) {
|
|
NewElts = malloc(NewCapacityInBytes);
|
|
if (NewElts == nullptr)
|
|
throw std::bad_alloc();
|
|
|
|
// Copy the elements over. No need to run dtors on PODs.
|
|
memcpy(NewElts, this->BeginX, CurSizeBytes);
|
|
} else {
|
|
// If this wasn't grown from the inline copy, grow the allocated space.
|
|
NewElts = realloc(this->BeginX, NewCapacityInBytes);
|
|
if (NewElts == nullptr)
|
|
throw std::bad_alloc();
|
|
}
|
|
|
|
this->EndX = (char*)NewElts + CurSizeBytes;
|
|
this->BeginX = NewElts;
|
|
this->CapacityX = (char*)this->BeginX + NewCapacityInBytes;
|
|
}
|
|
|
|
} // namespace c10
|