xref: /aosp_15_r20/external/pytorch/c10/core/Storage.cpp (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1 #include <c10/core/RefcountedDeleter.h>
2 #include <c10/core/Storage.h>
3 
4 namespace c10 {
5 
isSharedStorageAlias(const Storage & storage0,const Storage & storage1)6 bool isSharedStorageAlias(const Storage& storage0, const Storage& storage1) {
7   c10::DeleterFnPtr deleter_expected = &c10::refcounted_deleter;
8   c10::DeleterFnPtr deleter0 = storage0.data_ptr().get_deleter();
9   c10::DeleterFnPtr deleter1 = storage1.data_ptr().get_deleter();
10 
11   if ((deleter0 != deleter_expected) || (deleter1 != deleter_expected)) {
12     return false;
13   }
14 
15   return storage0.data_ptr().get_context() == storage1.data_ptr().get_context();
16 }
17 
18 } // namespace c10
19