forked from tinygrad/tinygrad
remove stale tests
This commit is contained in:
Vendored
-13
@@ -1,13 +0,0 @@
|
||||
#!/usr/bin/env python3
|
||||
from examples.hlb_cifar10 import SpeedyResNet, fetch_batch
|
||||
from examples.hlb_cifar10_torch import SpeedyResNet as SpeedyResNetTorch
|
||||
from extra.datasets import fetch_cifar
|
||||
from test.models.test_end2end import compare_tiny_torch
|
||||
|
||||
if __name__ == "__main__":
|
||||
X_test, Y_test = fetch_cifar(train=False)
|
||||
X, Y = fetch_batch(X_test, Y_test, 32)
|
||||
print(X.shape, Y.shape)
|
||||
model = SpeedyResNet()
|
||||
model_torch = SpeedyResNetTorch()
|
||||
compare_tiny_torch(model, model_torch, X, Y)
|
||||
Vendored
-17
@@ -1,17 +0,0 @@
|
||||
import weakref
|
||||
import numpy as np
|
||||
from tinygrad.tensor import Tensor, Device
|
||||
Device.DEFAULT = "METAL"
|
||||
|
||||
if __name__ == "__main__":
|
||||
t = Tensor.zeros(3).realize()
|
||||
wt = weakref.ref(t.lazydata.realized)
|
||||
n = t.numpy()
|
||||
t += 1
|
||||
n2 = t.numpy()
|
||||
print(wt)
|
||||
del t
|
||||
print(wt)
|
||||
print(n, n.base, n.base.base)
|
||||
print(n2, n2.base, n2.base.base)
|
||||
assert wt() is not None
|
||||
Reference in New Issue
Block a user