Skip to content

Commit

Permalink
Update the bindings.
Browse files Browse the repository at this point in the history
  • Loading branch information
LaurentMazare committed Jul 15, 2024
1 parent e26845d commit 16c4bb6
Show file tree
Hide file tree
Showing 6 changed files with 843 additions and 466 deletions.
3 changes: 2 additions & 1 deletion gen/gen.ml
Original file line number Diff line number Diff line change
Expand Up @@ -92,6 +92,7 @@ let excluded_prefixes =
; "_amp_foreach"
; "_nested_tensor"
; "_fused_adam"
; "_fused_adagrad"
; "sym_"
; "_fused_sgd"
]
Expand Down Expand Up @@ -590,7 +591,7 @@ let write_cpp funcs filename =
let pc s = p out_cpp s in
let ph s = p out_h s in
pc "// THIS FILE IS AUTOMATICALLY GENERATED, DO NOT EDIT BY HAND!";
pc "#include \"%s.h\"" (Caml.Filename.basename filename);
pc "#include \"%s.h\"" (Stdlib.Filename.basename filename);
pc "";
ph "// THIS FILE IS AUTOMATICALLY GENERATED, DO NOT EDIT BY HAND!";
ph "#include \"torch_api.h\"";
Expand Down
Loading

0 comments on commit 16c4bb6

Please sign in to comment.