MCPcopy Create free account
hub / github.com/AIS-SNU/Smart-Infinity / debug_param2name_id_shape_device

Function debug_param2name_id_shape_device

deepspeed/utils/debug.py:53–54  ·  view source on GitHub ↗
(param)

Source from the content-addressed store, hash-verified

51
52
53def debug_param2name_id_shape_device(param):
54 return f"name={debug_param2name(param)} id={param.ds_id} shape={param.data.shape} device={param.device}"
55
56
57def debug_param2name_id_numel(param):

Callers 2

partitionMethod · 0.90
partition_gradientsMethod · 0.90

Calls 1

debug_param2nameFunction · 0.85

Tested by

no test coverage detected