tech.compute.tvm.driver — tvm-clj 4.3

Additional protocols for tvm drivers, devices, and streams. Centralized registring of drivers allowing a symbolic name->driver table.

acceptable-tvm-device-buffer? clj

(acceptable-tvm-device-buffer? item)

acceptable-tvm-host-buffer? clj

(acceptable-tvm-host-buffer? item)

Things that pass this will work in copy-host->device.

has-byte-offset? clj

(has-byte-offset? buffer)

PTVMDriver clj protocol

->module clj

(->module driver sched-data-seq options)

device-id->device clj

(device-id->device driver device-id)

gpu-scheduling? clj

(gpu-scheduling? driver)

scalar-datatype->device-datatype clj

(scalar-datatype->device-datatype driver scalar-datatype)

schedule-injective! clj

(schedule-injective! driver stage compute-op options)

PTVMStream clj protocol

call-function clj

(call-function stream fn arg-list)