fix import of truncate (#7157)

truncate was moved to dtype
This commit is contained in:
chenyu
2024-10-18 18:41:41 -04:00
committed by GitHub
parent 54c6a317f8
commit 11beb67400
2 changed files with 4 additions and 4 deletions

View File

@@ -4,10 +4,10 @@
# this is the (living) definition of uops
from typing import Tuple, List, Optional, Any, Dict
import pickle, base64, itertools, time, struct
from tinygrad.dtype import DType, dtypes, ImageDType
from tinygrad.dtype import DType, dtypes, ImageDType, truncate
from tinygrad.helpers import all_same, getenv, flatten
from tinygrad.device import Compiled, Compiler, Allocator
from tinygrad.ops import BinaryOps, TernaryOps, exec_alu, truncate, UOps, UOp
from tinygrad.ops import BinaryOps, TernaryOps, exec_alu, UOps, UOp
from tinygrad.renderer import Renderer
from tinygrad.renderer.cstyle import CUDARenderer, MetalRenderer, AMDRenderer, IntelRenderer, ClangRenderer