| 37 | } |
| 38 | |
| 39 | operation gpu_allocation_model::preallocate(const shape& s, const std::string& id) const |
| 40 | { |
| 41 | return make_op("hip::hip_allocate_memory", {{"shape", to_value(s)}, {"id", id}}); |
| 42 | } |
| 43 | |
| 44 | std::string gpu_allocation_model::copy() const { return "hip::copy"; } |
| 45 |