tech.compute — techascent/tech.compute 4.46

->device clj

(->device item)

Generically get a device from a thing

->driver clj

(->driver item)

Generically get a driver from a thing

->stream clj

(->stream item)

Generically get a stream from a thing

allocate-device-buffer clj

(allocate-device-buffer device elem-count elem-type & {:as options})

Allocate a device buffer. This is the generic unit of data storage used for computation. No options at this time.

allocate-host-buffer clj

(allocate-host-buffer
  driver
  elem-count
  elem-type
  &
  {:keys [usage-type] :or {usage-type :one-time} :as options})

Allocate a host buffer. Usage type gives a hint as to the intended usage of the buffer.

copy-device->device clj

(copy-device->device dev-a
                     dev-a-off
                     dev-b
                     dev-b-off
                     elem-count
                     &
                     {:keys [stream] :as options})

Copy from one device to another. If no stream is provided then the destination buffer's device's default stream is used.

create-stream clj

(create-stream device)

Create a stream of execution. Streams are indepenent threads of execution. They can be synchronized with each other and the main thread using events.

default-device clj

(default-device driver)

default-stream clj

(default-stream device)

All devices must have a default stream whether they support create or not.

device->device-copy-compatible? clj

(device->device-copy-compatible? src-device dst-device)

driver clj

(driver driver-name)

Do a registry lookup to find a driver by its name.

driver-name clj

(driver-name driver)

driver-names clj

(driver-names)

Get the names of the registered drivers.

get-devices clj

(get-devices driver)

supports-create-stream? clj

(supports-create-stream? device)

Does this device support create-stream?

sync-with-host clj

(sync-with-host)

(sync-with-host stream)

Block host until stream's queue is finished executing

sync-with-stream clj

(sync-with-stream src-stream dst-stream & [options])

Create an event in src-stream's execution queue, then have dst stream wait on that event. This allows dst-stream to ensure src-stream has reached a certain point of execution before continuing. Both streams must be of the same driver.