tvm_tirx.tirx.script.free_workspace

Contents

tvm_tirx.tirx.script.free_workspace#

tvm_tirx.tirx.script.free_workspace(device_type, device_id, ptr, *, ty=None, loc=UNKNOWN_LOC)#

Backend function to free 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.

  • ptr (Var) – The result allocated space pointer.

  • loc (Location)

Returns:

call (Expr) – The call expression.

Contents