Function FreeTensor

gpu.hpp:705–716

* @brief Frees a tensor resource and updates the tensor pool. * * Only needed if the use case requires manually managing resource lifetimes of * GPU tensors. For simple use cases, the TensorPool destructor will * automatically free all tensors. * * @param[in] pool TensorPool instance to manage the tensor * @param[in] tensor Tensor instance to free * * @code * FreeTensor(pool, tensor); *

 * @endcode
 */
inline void FreeTensor(TensorPool &pool, Tensor tensor) {
  if (tensor.data.buffer) {
    wgpuBufferRelease(tensor.data.buffer);
  } else {
    LOG(kDefLog, kWarn, "Tried to free tensor with null buffer");
  }
  if (pool.data.find(tensor.data.buffer) != pool.data.end()) {
    pool.data.erase(tensor.data.buffer);
  } else {
    LOG(kDefLog, kWarn, "Tried to free tensor that was not in pool");
  }
}

/**
 * @brief Destructor for TensorPool which frees all tensors in the pool.