File tree Expand file tree Collapse file tree 1 file changed +4
-8
lines changed Expand file tree Collapse file tree 1 file changed +4
-8
lines changed Original file line number Diff line number Diff line change @@ -30,18 +30,16 @@ namespace aoti {
3030using  executorch::runtime::Error;
3131using  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+ 
3337extern  " C" 
3438
3539//  Common AOTI type aliases
3640using  AOTIRuntimeError = Error;
3741using  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)
4644AOTI_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
 
 
   
 
     
   
   
          
    
    
     
    
      
     
     
    You can’t perform that action at this time.
  
 
    
  
    
      
        
     
       
      
     
   
 
    
    
  
 
  
 
     
    
0 commit comments