Skip to content

Commit d164ae8

Browse files
committed
Fix so it works with ppx_minidebug disabled at compile time
1 parent a2d201e commit d164ae8

File tree

4 files changed

+16
-11
lines changed

4 files changed

+16
-11
lines changed

arrayjit/lib/backend_impl.ml

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -66,6 +66,8 @@ module No_device_buffer_and_copying () :
6666
let sexp_of_void_buffer_ptr (p : void_buffer_ptr) =
6767
Sexp.Atom (Ctypes_value_printing_stubs.string_of_pointer p)
6868

69+
let () = ignore sexp_of_void_buffer_ptr
70+
6971
let%track7_sexp memcpy ~(dst : void_buffer_ptr) ~(src : void_buffer_ptr) ~(size_in_bytes : int) :
7072
unit =
7173
if Ctypes_ptr.Fat.compare dst src <> 0 then

arrayjit/lib/tnode.ml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -794,4 +794,4 @@ let%debug_sexp log_accessible_headers ?(pred = fun _ -> true) () =
794794
Registry.fold (fun arr acc -> if pred arr then (arr.id, header arr) :: acc else acc) registry []
795795
in
796796
List.sort results ~compare:(fun (a, _) (b, _) -> compare_int a b)
797-
|> List.iter ~f:(fun (_, header) -> [%log header])
797+
|> List.iter ~f:(fun (_, _header) -> [%log _header])

arrayjit/lib/utils.ml

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -431,8 +431,11 @@ let _get_local_debug_runtime = get_local_debug_runtime
431431

432432
let%diagn_sexp set_log_level level =
433433
settings.log_level <- level;
434-
Debug_runtime.log_level := level;
435-
[%log "Set log_level to", (level : int)]
434+
[%log
435+
"Set log_level to",
436+
(Debug_runtime.log_level := level;
437+
level
438+
: int)]
436439

437440
let restore_settings () =
438441
set_log_level original_log_level;
@@ -713,9 +716,9 @@ let header_sep =
713716
compile (seq [ str " "; opt any; str "="; str " " ])
714717

715718
let%diagn_sexp log_trace_tree _logs =
716-
let sep s = String.concat ~sep:"\n" @@ String.split ~on:'$' s in
717719
[%log_block
718720
"trace tree";
721+
let sep s = String.concat ~sep:"\n" @@ String.split ~on:'$' s in
719722
let rec loop = function
720723
| [] -> []
721724
| line :: more when String.is_empty line -> loop more

lib/tensor.ml

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -722,14 +722,14 @@ let to_printbox ?single_node ?embedded_only ?entries_per_axis ?(with_id = false)
722722
~with_grad t
723723
|> PrintBox_utils.reformat_dag depth
724724

725-
let log_debug_info ~from_log_level t =
726-
let%diagn_sexp log_child { subtensor = _subtensor; embedded = _embedded } =
727-
[%logN_block
728-
from_log_level
729-
((if _embedded then "Embedded " else "Non-embedded ") ^ Tn.debug_name _subtensor.value);
730-
Tn.log_debug_info ~from_log_level _subtensor.value]
731-
in
725+
let%debug_sexp log_debug_info ~from_log_level t =
732726
[%diagn_sexp
727+
let%diagn_sexp log_child { subtensor = _subtensor; embedded = _embedded } =
728+
[%logN_block
729+
from_log_level
730+
((if _embedded then "Embedded " else "Non-embedded ") ^ Tn.debug_name _subtensor.value);
731+
Tn.log_debug_info ~from_log_level _subtensor.value]
732+
in
733733
[%logN_block
734734
from_log_level ("Tensor " ^ Tn.dims_to_string t.value);
735735
Tn.log_debug_info ~from_log_level t.value;

0 commit comments

Comments
 (0)