MCPcopy
hub / github.com/NVIDIA/TensorRT-LLM / numel

Method numel

triton_kernels/tensor.py:149–150  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

147 return self.storage.data.data_ptr()
148
149 def numel(self):
150 return self.storage.data.numel()
151
152 def element_size(self):
153 return bitwidth(self.dtype) // 8

Callers 14

matmul_ogsFunction · 0.95
assert_closeFunction · 0.45
forwardMethod · 0.45
__post_init__Method · 0.45
matmul_launch_metadataFunction · 0.45
alloc_fnFunction · 0.45
downcast_to_mxfpFunction · 0.45
swiglu_launch_metadataFunction · 0.45
run_testFunction · 0.45
fmha_bmark.pyFile · 0.45
_validate_non_emptyFunction · 0.45

Calls

no outgoing calls

Tested by 2

assert_closeFunction · 0.36
run_testFunction · 0.36