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

move frontend dir to nn [pr] (#12470)

Browse Source
This commit is contained in:
George Hotz
2025-10-07 10:42:22 +08:00
committed by GitHub
parent f664bcc8bd
commit 0f25b4b289
22 changed files with 18 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
extra/huggingface_onnx/run_models.py
View File

@@ -1,7 +1,7 @@
import onnx, yaml, tempfile, time, argparse, json
from pathlib import Path
from typing import Any
from tinygrad.frontend.onnx import OnnxRunner
from tinygrad.nn.onnx import OnnxRunner
from extra.onnx_helpers import validate, get_example_inputs
from extra.huggingface_onnx.huggingface_manager import DOWNLOADS_DIR, snapshot_download_with_retry

2
extra/onnx_helpers.py
View File

@@ -1,6 +1,6 @@
from tinygrad import Tensor
from tinygrad.tensor import _to_np_dtype
from tinygrad.frontend.onnx import OnnxRunner, OnnxValue
from tinygrad.nn.onnx import OnnxRunner, OnnxValue
import numpy as np
import onnxruntime as ort

2
extra/torch_backend/test_inplace.py
View File

@@ -1,6 +1,6 @@
import unittest
import torch
import tinygrad.frontend.torch
import tinygrad.nn.torch
torch.set_default_device("tiny")
import numpy as np

2
extra/torch_backend/test_multigpu.py
View File

@@ -1,7 +1,7 @@
import unittest
from tinygrad.helpers import getenv
import torch
import tinygrad.frontend.torch
import tinygrad.nn.torch
torch.set_default_device("tiny")
import numpy as np