TypeError in limit_dims_to_max method of Kernel.py

[JoeruCodes]

The lambda function of reshape_and_permute expects a lambda function which takes in a List[sint] as input. But the lambda function here takes x as input which is then used as an input for limit_size method which expects a List[int]

tinygrad/tinygrad/codegen/kernel.py

Line 330 in 6c7df14

self.reshape_and_permute(lambda x: self._limit_size(x, tmp + [math.inf] * (len(self.full_shape)-len(tmp))), None)

When I looked into the implementation of reshape_and_permute method, this lambda function is given a List[sint] as input from the shape method of shapetracker

[JoeruCodes]

changing x: Tuple[int] to x: Tuple[sint] or an assertion should do the job...

tinygrad/tinygrad/codegen/kernel.py

Line 312 in 6c7df14

def _limit_size(self, x: Tuple[int], max_size: List[Union[int,float]]) -> Tuple[int, ...]: