MCPcopy
hub / github.com/EleutherAI/gpt-neox / size

Method size

megatron/data/indexed_dataset.py:204–205  ·  view source on GitHub ↗
(self, index)

Source from the content-addressed store, hash-verified

202 return self.sizes[index]
203
204 def size(self, index):
205 return self.sizes[index]
206
207 @staticmethod
208 def exists(path):

Callers 15

split_into_partitionsFunction · 0.80
merge_partitionsFunction · 0.80
concat_partitionsFunction · 0.80
mainFunction · 0.80
get_ltor_masks_and_position_idsFunction · 0.80
filter_logitsFunction · 0.80
stream_tokensFunction · 0.80
get_batch_pipeFunction · 0.80
make_sparseMethod · 0.80
_orthogonalFunction · 0.80
forwardMethod · 0.80
forward_fused_softmaxMethod · 0.80

Calls

no outgoing calls

Tested by 5

test_fused_softmaxFunction · 0.64
test_fused_upper_triangle_mask_softmaxFunction · 0.64
test_layer_normFunction · 0.64
test_fused_softmaxFunction · 0.64
test_fused_upper_triangle_mask_softmaxFunction · 0.64