Skip to content

Commit 2255234

Browse files
committed
Fix documentation links
Signed-off-by: Lukasz Stafiniak <lukstafi@gmail.com>
1 parent 5e7759c commit 2255234

File tree

4 files changed

+16
-12
lines changed

4 files changed

+16
-12
lines changed

arrayjit/lib/backend_impl.ml

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -141,10 +141,11 @@ struct
141141
{ stream = parent.stream; parent = Some parent; ctx_arrays; finalized = Atomic.make false }
142142
end
143143

144-
(** Parts shared by backend implementations excluding what's already in {!Backend_any_common},
145-
except for {!Buffer} which is duplicated for technical reasons. *)
144+
(** Parts shared by backend implementations excluding what's already in
145+
{!Backend_intf.Backend_any_common}, except for {!Backend_intf.Buffer} which is duplicated for
146+
technical reasons. *)
146147
module type Backend_impl_common = sig
147-
include Buffer
148+
include Backend_intf.Buffer
148149

149150
val use_host_memory : (unit Ctypes.ptr -> buffer_ptr) option
150151
(** If not [None], the backend will read from and write to the host memory directly whenever
@@ -197,10 +198,11 @@ module type No_buffer_retrieval_or_syncing = sig
197198
include Backend_device_common with type buffer_ptr := buffer_ptr
198199

199200
val from_host : dst_ptr:buffer_ptr -> dst:context -> Ndarray.t -> unit
200-
(** Like {!Backend.from_host}, but without synchronization and buffer retrieval. *)
201+
(** Like {!Backend_intf.Backend.from_host}, but without synchronization and buffer retrieval. *)
201202

202203
val to_host : src_ptr:buffer_ptr -> src:context -> Ndarray.t -> unit
203-
(** Like {!Backend.to_host}, but without synchronization events and buffer retrieval. *)
204+
(** Like {!Backend_intf.Backend.to_host}, but without synchronization events and buffer retrieval.
205+
*)
204206

205207
val device_to_device :
206208
Tnode.t ->
@@ -210,8 +212,8 @@ module type No_buffer_retrieval_or_syncing = sig
210212
src_ptr:buffer_ptr ->
211213
src:context ->
212214
unit
213-
(** Like {!Backend.device_to_device}, but without synchronization events and buffer retrieval.
214-
Raises [Invalid_argument] if [into_merge_buffer = No] and [dst_ptr = None]. *)
215+
(** Like {!Backend_intf.Backend.device_to_device}, but without synchronization events and buffer
216+
retrieval. Raises [Invalid_argument] if [into_merge_buffer = No] and [dst_ptr = None]. *)
215217
end
216218

217219
(** An intermediate stage for converting {!Lowered_no_device_backend} backends into

arrayjit/lib/backend_intf.ml

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -199,8 +199,9 @@ module type Device = sig
199199
(** Returns a context without a parent. *)
200200

201201
val make_child : ?ctx_arrays:ctx_arrays -> context -> context
202-
(** Returns a context with the same {!field:stream}, and {!field:ctx_arrays} if omitted, as the
203-
given context's, which is also the {!field:parent}. *)
202+
(** Returns a context with the same {!field:Backend_intf.context.stream}, and
203+
{!field:Backend_intf.context.ctx_arrays} if omitted, as the given context's, which is also the
204+
{!field:Backend_intf.context.parent}. *)
204205

205206
val get_name : stream -> string
206207
end
@@ -290,7 +291,8 @@ module type Backend_device_common = sig
290291
val num_devices : unit -> int
291292

292293
val suggested_num_streams : device -> int
293-
(** The optimal number of streams for the given device to follow the {!type:config} strategy. *)
294+
(** The optimal number of streams for the given device to follow the {!type:Backend_intf.config}
295+
strategy. *)
294296

295297
val new_stream : device -> stream
296298
end

arrayjit/lib/backends.mli

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,5 +19,5 @@ val finalize :
1919

2020
val fresh_backend : ?backend_name:string -> unit -> (module Backend_intf.Backend)
2121
(** Creates a new backend corresponding to [backend_name], or if omitted, selected via the global
22-
[backend] setting. It should be safe to call {!Tensor.unsafe_reinitialize} before
22+
[backend] setting. It should be safe to reinitialize the tensor system before
2323
[fresh_backend]. *)

lib/shape.mli

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,7 @@ type transpose_type =
8383
| Batch_slice of Arrayjit.Indexing.static_symbol (** Removes the leftmost batch axis. *)
8484
[@@deriving equal, sexp]
8585

86-
(** If you miss expressivity here, leave a note on {!{https://github.com/ahrefs/ocannl/issues/305}issue 305}. *)
86+
(** If you miss expressivity here, leave a note on {{:https://github.com/ahrefs/ocannl/issues/305}issue 305}. *)
8787
type ternary_type =
8888
| Pointwise_tern (** As in the operation [Where]. *)
8989
| Compose_accumulate (** As in the operation [FMA]. *)

0 commit comments

Comments
 (0)