1 files changed, 7 insertions, 2 deletions

diff --git a/src/syscalls.c b/src/syscalls.c
index b6f6c5c..b0fee23 100644
--- a/src/syscalls.c
+++ b/src/syscalls.c
@@ -89,7 +89,12 @@ int null_call(cycles_t *timestamp)
         return syscall(__NR_null_call, timestamp);
 }
 
-int register_nv_device(pid_t pid, int nv_device_id)
+int register_nv_device(int nv_device_id)
 {
-        return syscall(__NR_register_nv_device, pid, nv_device_id);
+        return syscall(__NR_register_nv_device, nv_device_id, 1);
+}
+
+int unregister_nv_device(int nv_device_id)
+{
+        return syscall(__NR_register_nv_device, nv_device_id, 0);
 }