diff --git a/tritonparse/reproducer/placeholder_replacer.py b/tritonparse/reproducer/placeholder_replacer.py index 2749490..aff584a 100644 --- a/tritonparse/reproducer/placeholder_replacer.py +++ b/tritonparse/reproducer/placeholder_replacer.py @@ -5,7 +5,6 @@ from tritonparse.reproducer.function_extractor import extract_utility_functions from tritonparse.reproducer.ingestion.ndjson import ContextBundle - from tritonparse.reproducer.templates.utils import ( _disable_triton_autotune, get_function_source, diff --git a/tritonparse/reproducer/utils.py b/tritonparse/reproducer/utils.py index 1ad696e..fffa2aa 100644 --- a/tritonparse/reproducer/utils.py +++ b/tritonparse/reproducer/utils.py @@ -12,7 +12,6 @@ import torch import triton.language as tl - from tritonparse.tools.load_tensor import load_tensor from tritonparse.tp_logger import logger