From cd79a904c5d8ab36daf1804e69f214941dfc8f47 Mon Sep 17 00:00:00 2001 From: chenyu Date: Fri, 20 Dec 2024 13:19:48 -0500 Subject: [PATCH] add back explicit `dict[DType, str]` in ptx [pr] (#8352) --- tinygrad/renderer/ptx.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tinygrad/renderer/ptx.py b/tinygrad/renderer/ptx.py index e7fb4244d0..c7eaf6f32f 100644 --- a/tinygrad/renderer/ptx.py +++ b/tinygrad/renderer/ptx.py @@ -142,7 +142,7 @@ class PTXRenderer(Renderer): dtypes.uint8: "u16", dtypes.uint16: "u16", dtypes.uint32: "u32", dtypes.uint64: "u64", dtypes.float16: "f16", dtypes.float32: "f32", dtypes.float64: "f64", dtypes.bool: "pred" } - mem_types = {**types, dtypes.int8: "s8", dtypes.uint8: "u8", dtypes.bool: "u8", dtypes.float16: "b16"} + mem_types: dict[DType, str] = {**types, dtypes.int8: "s8", dtypes.uint8: "u8", dtypes.bool: "u8", dtypes.float16: "b16"} def render_kernel(self, kernel, function_name, bufs, regs) -> str: def fmt(line): return line if line[0]=="$" else "\t" + line.replace(" ", "\t" if len(line.split(" ")[0]) > 7 else "\t\t", 1)