MCPcopy Create free account
hub / github.com/deepspeedai/DeepSpeed / getGradBuffer

Method getGradBuffer

csrc/includes/deepcompile.h:294–294  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

292 return ds_tensor_;
293 }
294 at::Tensor getGradBuffer() const { return grad_buffer_; }
295 void setGradBuffer(at::Tensor grad_buffer, int64_t offset)
296 {
297 grad_buffer_ = grad_buffer;

Callers 3

flushReduceBucketMethod · 0.80
flushReduceBucketMethod · 0.80
flushReduceBucketMethod · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected