Skip to content

Commit 8b46964

Browse files
committed
move vectors
1 parent 9dc28d9 commit 8b46964

File tree

1 file changed

+4
-8
lines changed

1 file changed

+4
-8
lines changed

backends/aoti/common_shims.h

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -30,18 +30,16 @@ namespace aoti {
3030
using executorch::runtime::Error;
3131
using executorch::runtime::etensor::Tensor;
3232

33+
// Global storage for tensor metadata
34+
extern std::unordered_map<Tensor*, std::vector<int64_t>> tensor_to_sizes;
35+
extern std::unordered_map<Tensor*, std::vector<int64_t>> tensor_to_strides;
36+
3337
extern "C" {
3438

3539
// Common AOTI type aliases
3640
using AOTIRuntimeError = Error;
3741
using AOTITorchError = Error;
3842

39-
// Global storage for tensor metadata
40-
extern AOTI_SHIM_EXPORT std::unordered_map<Tensor*, std::vector<int64_t>>
41-
tensor_to_sizes;
42-
extern AOTI_SHIM_EXPORT std::unordered_map<Tensor*, std::vector<int64_t>>
43-
tensor_to_strides;
44-
4543
// Attribute-related operations (memory-irrelevant)
4644
AOTI_SHIM_EXPORT AOTITorchError aoti_torch_get_data_ptr(
4745
Tensor* tensor,
@@ -100,5 +98,3 @@ AOTI_SHIM_EXPORT void cleanup_tensor_metadata();
10098
} // namespace aoti
10199
} // namespace backends
102100
} // namespace executorch
103-
104-
#undef AOTI_SHIM_EXPORT

0 commit comments

Comments
 (0)