Additional protocols for tvm drivers, devices, and streams. Centralized registring of drivers allowing a symbolic name->driver table.
Additional protocols for tvm drivers, devices, and streams. Centralized registring of drivers allowing a symbolic name->driver table.
(acceptable-tvm-host-buffer? item)
Things that pass this will work in copy-host->device.
Things that pass this will work in copy-host->device.
(->module driver sched-data-seq options)
(device-id->device driver device-id)
(gpu-scheduling? driver)
(scalar-datatype->device-datatype driver scalar-datatype)
(schedule-injective! driver stage compute-op options)
cljdoc is a website building & hosting documentation for Clojure/Script libraries
× close