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

Method getEvent

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

Source from the content-addressed store, hash-verified

227 }
228
229 std::shared_ptr<at::cuda::CUDAEvent> getEvent(at::ScalarType scalar_type)
230 {
231 assert(hasKey(current_buffer_events_, scalar_type));
232 return current_buffer_events_.at(scalar_type);
233 }
234
235 void clear()
236 {

Callers 1

reduceGradMethod · 0.80

Calls 1

hasKeyFunction · 0.85

Tested by

no test coverage detected