From 793ace530ea4b2627744b47d7ce7d8f9ed0b2263 Mon Sep 17 00:00:00 2001 From: qazal <77887910+Qazalin@users.noreply.github.com> Date: Fri, 8 Aug 2025 22:07:35 +0800 Subject: [PATCH] update amd_uop_matmul.py import (#11581) Using this for testing SQTT --- extra/gemm/amd_uop_matmul.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/extra/gemm/amd_uop_matmul.py b/extra/gemm/amd_uop_matmul.py index 9807326dbe..d217edb8c5 100644 --- a/extra/gemm/amd_uop_matmul.py +++ b/extra/gemm/amd_uop_matmul.py @@ -2,11 +2,11 @@ from tinygrad import Tensor, Device, Context, GlobalCounters, dtypes from tinygrad.uop.ops import UOp, Ops, KernelInfo, graph_rewrite, AxisType, PatternMatcher, UPat from tinygrad.engine.realize import CompiledRunner, ExecItem, get_program from tinygrad.dtype import AddrSpace -from tinygrad.schedule.kernelize import merge_views, view_left from tinygrad.helpers import getenv, colored, prod, unwrap from tinygrad.shape.shapetracker import ShapeTracker, View from tinygrad.shape.view import strides_for_shape from tinygrad.codegen.opt.kernel import axis_colors +from tinygrad.codegen.opt.swizzler import merge_views, view_left def to_colored(full_shape, axis_types): return '_'.join([colored(str(s), axis_colors[at]) for s,at in zip(full_shape, axis_types)])