| // This file creates a fake allocator that just throws exceptions if |
| // state passed to the allocator is the std::function<void(void*)> called |
| // when the blob is release by ATen |
| static cpu_fixed_malloc(void *, ptrdiff_t) { |
| runtime_error("attempting to resize a tensor view of an external blob"); |
| static cpu_fixed_realloc(void *, void*, ptrdiff_t) { |
| runtime_error("attempting to resize a tensor view of an external blob"); |
| static cpu_fixed_free(void * state, void * allocation) { |
| auto on_release = static_cast<std::function<void(void*)>*>(state); |
| (*on_release)(allocation); |
| static THAllocator CPU_fixed_allocator = |
| { cpu_fixed_malloc, cpu_fixed_realloc, cpu_fixed_free }; |