tvm_tirx.tirx.script.alloc_workspace

tvm_tirx.tirx.script.alloc_workspace#

tvm_tirx.tirx.script.alloc_workspace(device_type, device_id, nbytes, dtype_code_hint, dtype_bits_hint, *, ty=None, loc=UNKNOWN_LOC)#

Backend function to allocate temporal workspace

Parameters:

  • device_type (int) – The device type which the space will be allocated.

  • device_id (int) – The device id which the space will be allocated.

  • nbytes (int) – The size of the space requested.

  • dtype_code_hint (int) – The type code of the array elements. Only used in certain backends such as OpenGL.

  • dtype_bits_hint (int) – The type bits of the array elements. Only used in certain backends such as OpenGL.

  • loc (Location)

Returns:

call (Expr) – The call expression.