github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

Deprecated includes removed

Browse Source
This commit is contained in:
Szymon Ożóg
2023-08-16 07:17:08 +02:00
parent 7616b27a5b
commit 7045109592
1 changed files with 4 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
tinygrad/runtime/ops_triton.py
View File

@@ -1,25 +1,17 @@
from __future__ import annotations
import hashlib
from weakref import WeakValueDictionary
from torch import float32
import numpy as np
import pycuda.autoprimaryctx # type: ignore # noqa: F401
import pycuda.driver as cuda # type: ignore
import triton # type: ignore # noqa: F401
import triton.language as tl # type: ignore # noqa: F401
from triton.compiler import compile as triton_compile
from typing import Any, Union, Tuple, Optional, Dict, List, Final, Callable
from tinygrad.ops import UnaryOps, BinaryOps, ReduceOps, LazyOp, Op, GlobalCounters, Compiled
from tinygrad.shape.shapetracker import ShapeTracker
from typing import Any, Tuple, Dict, List, Final, Callable
from tinygrad.ops import UnaryOps, BinaryOps, ReduceOps, Op, Compiled
from tinygrad.helpers import prod, DEBUG, dtypes, ImageDType
from tinygrad.runtime.ops_gpu import RawBufferCopyInOut
from tinygrad.runtime.ops_cuda import RawCUDABuffer
from tinygrad.runtime.lib import RawBuffer
from tinygrad.codegen.linearizer import LinearizerOptions, UOp, UOps, LocalBuffer
from tinygrad.codegen.linearizer import LinearizerOptions, UOp, UOps
from tinygrad.shape.symbolic import NumNode
class TritonProgram:
def __init__(self, name:str, prg:str, binary:bool=False):