Function custom_set0_kernel

test/unit/test_schedule_cache.py:7–8

(A:UOp, num:int)

from tinygrad.schedule import schedule_cache

def custom_set0_kernel(A:UOp, num:int) -> UOp:
  return A[0].set(num).sink(arg=KernelInfo(f"custom_set0_{num}"))

class TestScheduleCache(unittest.TestCase):
  def test_bound_variable_reuses_cache(self):