Function AllocateWorkspace

tensorflow/stream_executor/cuda/cuda_blas.cc:3323–3330

Helper function to allocate workspace.

// Helper function to allocate workspace.
port::Status AllocateWorkspace(void** workspace,
                               ScratchAllocator* scratch_allocator,
                               size_t num_bytes) {
  SE_ASSIGN_OR_RETURN(DeviceMemory<uint8> workspace_bytes,
                      scratch_allocator->AllocateBytes(num_bytes));
  *workspace = (void*)GpuMemoryMutable(&workspace_bytes);
  return port::Status::OK();
}

template<typename T>
blas::ComputationType ToComputationType();