MCPcopy
hub / github.com/tinygrad/tinygrad / store

Method store

extra/hcq2/ops_amd2.py:192–195  ·  view source on GitHub ↗
(self, x)

Source from the content-addressed store, hash-verified

190 self.sdma.SDMA_PKT_POLL_REGMEM_DW5_INTERVAL(0x04) | self.sdma.SDMA_PKT_POLL_REGMEM_DW5_RETRY_COUNT(0xfff))
191
192 def store(self, x):
193 fence_flags = self.sdma.SDMA_PKT_FENCE_HEADER_MTYPE(3) if self.dev.target[0] != 9 else 0
194 self.q(self.sdma.SDMA_OP_FENCE | fence_flags, *data64_le(self.get_dev_addr(x.src[0])), x.src[1])
195 self.q(self.sdma.SDMA_OP_TRAP, 0)
196
197 def timestamp(self, x):
198 self.q(self.sdma.SDMA_OP_TIMESTAMP | self.sdma.SDMA_PKT_TIMESTAMP_GET_HEADER_SUB_OP(self.sdma.SDMA_SUBOP_TIMESTAMP_GET_GLOBAL),

Callers 15

copyFunction · 0.45
hand_spec_kernel3Function · 0.45
block_128x128_gemmFunction · 0.45
hand_spec_tc_coresFunction · 0.45
amd_flash_attentionFunction · 0.45
matmul_kernelFunction · 0.45
generic_copyFunction · 0.45
compute_on_localsFunction · 0.45
custom_gemmFunction · 0.45
custom_uop_gemmFunction · 0.45
custom_gemm_bwFunction · 0.45
copyFunction · 0.45

Calls 3

qMethod · 0.95
data64_leFunction · 0.90
get_dev_addrMethod · 0.80

Tested by

no test coverage detected