Missing Context.sexp_of_t conversion

lukstafi · Oct 18, 2024 · 64e5820 · 64e5820
1 parent e2b6dce
commit 64e5820
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/cudajit.ml b/cudajit.ml
@@ -1087,6 +1087,11 @@ let no_stream =
 module Context = struct
   type t = cu_context
 
+  let sexp_of_t (ctx : t) =
+    Sexplib0.Sexp.Atom
+      ("@" ^ Unsigned.UInt64.to_hexstring @@ Unsigned.UInt64.of_string @@ Nativeint.to_string
+     @@ Ctypes.raw_address_of_ptr @@ Ctypes.to_voidp ctx)
+
   type flag =
     | SCHED_AUTO
     | SCHED_SPIN

diff --git a/cudajit.mli b/cudajit.mli
@@ -296,7 +296,7 @@ module Context : sig
 
   type flags = flag list [@@deriving sexp]
 
-  type t
+  type t [@@deriving sexp_of]
   (** See
       {{:https://docs.nvidia.com/cuda/cuda-driver-api/group__CUDA__TYPES.html#group__CUDA__TYPES_1gf9f5bd81658f866613785b3a0bb7d7d9}
         CUcontext}. *)