github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-10 07:28:15 -05:00
Code Issues Packages Projects Releases Wiki Activity

move image+kopt to features (#2015)

Browse Source 
* move image+kopt to features

* fix tests

* debug prints (unrelated)
This commit is contained in:
George Hotz
2023-10-07 15:41:08 -07:00
committed by GitHub
parent 44ed94ef5c
commit cea4cbfc7a
9 changed files with 239 additions and 227 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_schedule.py
View File

@@ -262,7 +262,7 @@ class TestSchedule(unittest.TestCase):
# this is the failing case in openpilot...it's very simple like this
@unittest.skip("failing in old lazy")
def test_image_conv_fusion(self):
from tinygrad.nn.image import image_conv2d
from tinygrad.features.image import image_conv2d
w1 = Tensor.empty(16, 16, 1, 1)
b1 = Tensor.empty(16)
w2 = Tensor.empty(16, 16, 1, 1)