From bd2a77552886b98c205af12f8d7d2d61247c4b27 Mon Sep 17 00:00:00 2001 From: Chenggang Zhao Date: Tue, 11 Mar 2025 13:26:10 +0800 Subject: [PATCH] Code format --- deep_gemm/jit/template.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deep_gemm/jit/template.py b/deep_gemm/jit/template.py index c983817..cdca4c4 100644 --- a/deep_gemm/jit/template.py +++ b/deep_gemm/jit/template.py @@ -1,9 +1,9 @@ import copy import ctypes import os +import torch from typing import Any, Dict, Iterable, Tuple -import torch # Name map for Python `eval` typename_map: Dict[Any, str] = {