| 54 | } |
| 55 | |
| 56 | void register_memory_management(VM* vm) { |
| 57 | vm_register_instruction_load(vm, ns(Instruction_DevMemAlloc), &load_alloc_device); |
| 58 | vm_register_instruction_call(vm, TinyNN_INST_DEV_MEM_ALLOC, &alloc_device_tensor); |
| 59 | vm_register_instruction_destruct(vm, TinyNN_INST_DEV_MEM_ALLOC, &destruct); |
| 60 | vm_register_instruction_load(vm, ns(Instruction_DevMemFree), &load_free_device); |
| 61 | vm_register_instruction_call(vm, TinyNN_INST_DEV_MEM_FREE, &free_device_tensor); |
| 62 | vm_register_instruction_destruct(vm, TinyNN_INST_DEV_MEM_FREE, &destruct); |
| 63 | } |
| 64 | #else |
| 65 | void register_memory_management(VM* vm) {} |
| 66 | #endif |
no test coverage detected