Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
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
flushReduceBucket
Method · 0.80
flushReduceBucket
Method · 0.80
flushReduceBucket
Method · 0.80
Calls
no outgoing calls
Tested by
no test coverage detected