github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-10 23:48:01 -05:00
Code Issues Packages Projects Releases Wiki Activity

touchup remove float() in cstyle render_const for float64 (#3962)

Browse Source
This commit is contained in:
chenyu
2024-03-27 16:08:28 -04:00
committed by GitHub
parent 27af37f2ad
commit 88b24df40a
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tinygrad/renderer/cstyle.py
View File

@@ -41,10 +41,10 @@ class CStyleLanguage(NamedTuple):
return f"{self.float4.replace('float4', self.render_dtype(var_dtype))}({','.join(x)})"
# returns a str expression of the const with the given type
def render_const(self, x:Union[float,int,bool], var_dtype) -> str:
def render_const(self, x:Union[float,int,bool], var_dtype:DType) -> str:
if math.isnan(x): val = "NAN"
elif math.isinf(x): val = ("-" if x < 0 else "") + "INFINITY"
elif var_dtype == dtypes.float64: val = f"{float(x)}"
elif var_dtype == dtypes.float64: val = f"{x}"
else: val = f"{x}f" if dtypes.is_float(var_dtype) else f"{x}" if dtypes.is_int(var_dtype) else f"{x}".lower()
return (self.render_cast([val]*var_dtype.count, var_dtype)
if var_dtype.count > 1 or var_dtype not in [dtypes.float, dtypes.int, dtypes.bool] else val)
@@ -91,7 +91,6 @@ class CStyleLanguage(NamedTuple):
def uops_to_cstyle(lang:CStyleLanguage, function_name:str, uops:UOpGraph) -> str:
kernel = []
bufs: List[Tuple[str, Tuple[DType, bool]]] = []
#pend_close = None
depth = 1
def kk(s): kernel.append(" "*depth+s)