github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-02-12 07:35:16 -05:00
Code Issues Packages Projects Releases Wiki Activity

relax atol, merge_view

Browse Source
This commit is contained in:
George Hotz
2023-03-03 07:48:44 -08:00
parent b9ce20c374
commit 8c475ea86a
2 changed files with 9 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
test/unit/test_shapetracker.py
View File

@@ -2,7 +2,7 @@
import unittest
import numpy as np
from tinygrad.helpers import prod
from tinygrad.shape import ShapeTracker, View, ZeroView
from tinygrad.shape import ShapeTracker, View, ZeroView, merge_views
from tinygrad.codegen.gpu import to_image_idx
def shapetracker_getitem(st, val):
@@ -75,6 +75,12 @@ class TestImageShapeTracker(unittest.TestCase):
offsets = [0,32,64]
"""
new_view = merge_views(
View((1, 66, 130, 32, 1, 1), (0, 4096, 32, 1, 0, 0), -4128),
View((64, 32, 8, 3, 3), (4160, 128, 4, 4160, 32), 0)
)
print(new_view)
st = ShapeTracker(shape=(64, 32, 8, 3, 3), views=[
View((1, 66, 130, 32, 1, 1), (0, 4096, 32, 1, 0, 0), -4128),
ZeroView((1, 64, 128, 32, 1, 1), ((0, 1), (-1, 65), (-1, 129), (0, 32), (0, 1), (0, 1))),