mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
33a5179269
Summary: Reland https://github.com/pytorch/pytorch/pull/151109 after fixing cutlass AOTI build issues. typedef is prone to name collision. Explicitly spell out the actual aten types, needed for the standalone AOTI codegen. Differential Revision: D74398762 Pull Request resolved: https://github.com/pytorch/pytorch/pull/153467 Approved by: https://github.com/jingsh, https://github.com/henrylhtsang, https://github.com/cyyever
17 lines
416 B
C++
17 lines
416 B
C++
#pragma once
|
|
|
|
#include <array>
|
|
#include <filesystem>
|
|
#include <optional>
|
|
|
|
#include <torch/csrc/inductor/aoti_runtime/interface.h>
|
|
#include <torch/csrc/inductor/aoti_runtime/model.h>
|
|
|
|
#include <c10/util/generic_math.h>
|
|
#include <torch/csrc/inductor/aoti_runtime/scalar_to_tensor.h>
|
|
|
|
// Round up to the nearest multiple of 64
|
|
[[maybe_unused]] inline int64_t align(int64_t nbytes) {
|
|
return (nbytes + 64 - 1) & -64;
|
|
}
|