mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
9665702c64
Summary: fix windows build broke in https://github.com/pytorch/pytorch/pull/156508 Test Plan: ci Rollback Plan: Differential Revision: D77080420 Pull Request resolved: https://github.com/pytorch/pytorch/pull/156565 Approved by: https://github.com/zhxchen17
14 lines
380 B
C++
14 lines
380 B
C++
#pragma once
|
|
|
|
#include <torch/nativert/executor/memory/LayoutPlannerAlgorithm.h>
|
|
|
|
namespace torch::nativert {
|
|
|
|
// lay out all tensors contiguously in memory
|
|
// this doesn't take into account lifetimes,
|
|
// it literally just puts them all next to each other
|
|
LayoutPlan BumpAllocationPlanner(
|
|
const std::vector<AllocationSpec>& allocation_specs);
|
|
|
|
} // namespace torch::nativert
|