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

move hcq device to runtime [pr] (#6879)

Browse Source 
* things that are only used in one place don't belong in helpers [pr]

* start moving hcq device [pr]

* fix paths
This commit is contained in:
George Hotz
2024-10-04 22:26:50 +08:00
committed by GitHub
parent 5be2bd18a6
commit 6b063450df
9 changed files with 563 additions and 554 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tinygrad/runtime/ops_nv.py
View File

@@ -2,8 +2,9 @@ from __future__ import annotations
import os, ctypes, contextlib, re, fcntl, functools, mmap, struct, array, decimal
from typing import Tuple, List, Any, cast, Union, Dict, Type
from dataclasses import dataclass
from tinygrad.device import HCQCompiled, HCQAllocator, HCQBuffer, HWCommandQueue, HWComputeQueue, HWCopyQueue, hcq_command, \
HCQArgsState, HCQProgram, HCQSignal, BufferOptions
from tinygrad.runtime.support.hcq import HCQCompiled, HCQAllocator, HCQBuffer, HWCommandQueue, HWComputeQueue, HWCopyQueue, hcq_command
from tinygrad.runtime.support.hcq import HCQArgsState, HCQProgram, HCQSignal
from tinygrad.device import BufferOptions
from tinygrad.helpers import getenv, mv_address, init_c_struct_t, to_mv, round_up, data64, data64_le, DEBUG, prod
from tinygrad.renderer.assembly import PTXRenderer
from tinygrad.renderer.cstyle import NVRenderer