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

Revert "gated native math in OpenCL (#6683)" (#6691)

Browse Source 
This reverts commit 2fe3eeed17.
This commit is contained in:
George Hotz
2024-09-24 08:48:10 +08:00
committed by GitHub
parent 382938ab41
commit b9e6d42a1f
3 changed files with 0 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
examples/openpilot/compile2.py
View File

@@ -6,7 +6,6 @@ sys.path.insert(0, str(pathlib.Path(__file__).parents[1]))
if "FLOAT16" not in os.environ: os.environ["FLOAT16"] = "1"
if "IMAGE" not in os.environ: os.environ["IMAGE"] = "2"
if "NOLOCALS" not in os.environ: os.environ["NOLOCALS"] = "1"
if "NATIVE_MATH" not in os.environ: os.environ["NATIVE_MATH"] = "1"
OPENPILOT_MODEL = "https://github.com/commaai/openpilot/raw/v0.9.4/selfdrive/modeld/models/supercombo.onnx"

1
examples/openpilot/compile3.py
View File

@@ -3,7 +3,6 @@ import numpy as np
if "FLOAT16" not in os.environ: os.environ["FLOAT16"] = "1"
if "IMAGE" not in os.environ: os.environ["IMAGE"] = "2"
if "NOLOCALS" not in os.environ: os.environ["NOLOCALS"] = "1"
if "NATIVE_MATH" not in os.environ: os.environ["NATIVE_MATH"] = "1"
from tinygrad import fetch, Tensor, TinyJit, Device, Context, GlobalCounters
from tinygrad.helpers import OSX, DEBUG, Timing

5
tinygrad/renderer/cstyle.py
View File

@@ -227,11 +227,6 @@ class ClangRenderer(CStyleLanguage):
class OpenCLRenderer(CStyleLanguage):
device = "GPU"
code_for_op = {**CStyleLanguage().code_for_op,
#UnaryOps.SQRT: lambda x,dtype: f"native_sqrt({x})", UnaryOps.RECIP: lambda x,dtype: f"native_recip({x})",
#UnaryOps.EXP2: lambda x,dtype: f"native_exp2({x})", UnaryOps.LOG2: lambda x,dtype: f"native_log2({x})",
UnaryOps.SIN: lambda x,dtype: f"native_sin({x})"} if getenv("NATIVE_MATH") else CStyleLanguage().code_for_op
# language options
kernel_prefix = "__kernel "
buffer_prefix = "__global "