From ca0dca35f7929ecbb47673bc682e289f2dec5c13 Mon Sep 17 00:00:00 2001 From: George Hotz <72895+geohot@users.noreply.github.com> Date: Thu, 17 Oct 2024 14:50:32 +0800 Subject: [PATCH] move ptx renderer [pr] (#7118) --- tinygrad/renderer/{assembly.py => ptx.py} | 0 tinygrad/runtime/ops_cuda.py | 2 +- tinygrad/runtime/ops_nv.py | 2 +- 3 files changed, 2 insertions(+), 2 deletions(-) rename tinygrad/renderer/{assembly.py => ptx.py} (100%) diff --git a/tinygrad/renderer/assembly.py b/tinygrad/renderer/ptx.py similarity index 100% rename from tinygrad/renderer/assembly.py rename to tinygrad/renderer/ptx.py diff --git a/tinygrad/runtime/ops_cuda.py b/tinygrad/runtime/ops_cuda.py index 33288073f1..ef01369133 100644 --- a/tinygrad/runtime/ops_cuda.py +++ b/tinygrad/runtime/ops_cuda.py @@ -4,7 +4,7 @@ from typing import Tuple, Optional, List from tinygrad.helpers import DEBUG, getenv, from_mv, init_c_var, init_c_struct_t from tinygrad.device import Compiled, BufferOptions, LRUAllocator from tinygrad.renderer.cstyle import CUDARenderer -from tinygrad.renderer.assembly import PTXRenderer +from tinygrad.renderer.ptx import PTXRenderer from tinygrad.runtime.autogen import cuda from tinygrad.runtime.support.compiler_cuda import cuda_disassemble, pretty_ptx, CUDACompiler, PTXCompiler, PTX if getenv("IOCTL"): import extra.nv_gpu_driver.nv_ioctl # noqa: F401 # pylint: disable=unused-import diff --git a/tinygrad/runtime/ops_nv.py b/tinygrad/runtime/ops_nv.py index c22c82a65f..17dd681638 100644 --- a/tinygrad/runtime/ops_nv.py +++ b/tinygrad/runtime/ops_nv.py @@ -6,7 +6,7 @@ from tinygrad.runtime.support.hcq import HCQCompiled, HCQAllocator, HCQBuffer, H 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.ptx import PTXRenderer from tinygrad.renderer.cstyle import NVRenderer from tinygrad.runtime.support.compiler_cuda import CUDACompiler, PTXCompiler, PTX, NVPTXCompiler, NVCompiler, nv_disassemble from tinygrad.runtime.autogen import nv_gpu, libc