MCPcopy
hub / github.com/tinygrad/tinygrad / update

Method update

tinygrad/runtime/support/nv/nvdev.py:24–24  ·  view source on GitHub ↗
(self, **kwargs)

Source from the content-addressed store, hash-verified

22 def write(self, _ini_val:int=0, **kwargs): self.nvdev.wreg(self.base + self.off, _ini_val | self.encode(**kwargs))
23
24 def update(self, **kwargs): self.write(self.read() & ~self.mask(*kwargs.keys()), **kwargs)
25
26 def mask(self, *names):
27 return functools.reduce(int.__or__, ((((1 << (self.fields[nm][1]-self.fields[nm][0] + 1)) - 1) << self.fields[nm][0]) for nm in names), 0)

Callers 14

gguf_loadFunction · 0.45
render_wmma_amdFunction · 0.45
__init_subclass__Method · 0.45
generate.pyFile · 0.45
get_state_dictFunction · 0.45
__call__Method · 0.45
run_branchFunction · 0.45
_rangesMethod · 0.45
deconstruct_functionFunction · 0.45
_resolve_depsMethod · 0.45
execute_hsMethod · 0.45
disable_ctx_reqMethod · 0.45

Calls 4

writeMethod · 0.95
readMethod · 0.95
maskMethod · 0.95
keysMethod · 0.80

Tested by

no test coverage detected