move metadata set to tensor [pr] (#9976)

* move metadata set to tensor [pr]

* only track that in tensor.py
This commit is contained in:
George Hotz
2025-04-22 12:30:35 +01:00
committed by GitHub
parent f6271515fe
commit e358e0a0c6
4 changed files with 13 additions and 11 deletions

View File

@@ -3,7 +3,8 @@ import numpy as np
import torch
import unittest, copy, mmap, random, math, array
from tinygrad import Tensor, Device, dtypes
from tinygrad.helpers import getenv, temp, _METADATA, mv_address
from tinygrad.tensor import _METADATA
from tinygrad.helpers import getenv, temp, mv_address
from extra.gradcheck import numerical_jacobian, jacobian, gradcheck
from hypothesis import given, settings, strategies as strat
from tinygrad.device import is_dtype_supported