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

Method stride

triton_kernels/tensor.py:143–144  ·  view source on GitHub ↗
(self, i=None)

Source from the content-addressed store, hash-verified

141 return self.storage.device
142
143 def stride(self, i=None):
144 return self.storage.data.stride() if i is None else self.storage.data.stride(i)
145
146 def data_ptr(self):
147 return self.storage.data.data_ptr()

Callers 15

matmul_ogsFunction · 0.95
topk_forwardFunction · 0.95
can_overflow_int32Function · 0.80
_canonicalize_storageFunction · 0.80
reduce_groupedFunction · 0.80
forwardMethod · 0.80
topk_backwardFunction · 0.80
forwardMethod · 0.80
forwardMethod · 0.80
compute_expt_dataFunction · 0.80
compactionFunction · 0.80
is_tma_compliantMethod · 0.80

Calls

no outgoing calls