[FRONTEND] cleaned up symbol names (#1782)

This commit is contained in:
Philippe Tillet
2023-06-14 18:55:32 -07:00
committed by GitHub
parent 71e21f5797
commit b24dc19741
10 changed files with 4 additions and 9 deletions

View File

@@ -265,7 +265,7 @@ setup(
"triton/_C",
"triton/common",
"triton/compiler",
"triton/debugger",
"triton/interpreter",
"triton/language",
"triton/language/extra",
"triton/ops",

View File

@@ -4,7 +4,7 @@ import torch
import triton
import triton.language as tl
from triton.debugger.debugger import program_ids_from_grid
from triton.interpreter.interpreter import program_ids_from_grid
def test_addition():

View File

@@ -18,7 +18,6 @@ from .runtime import (
)
from .runtime.jit import jit
from .compiler import compile, CompilationError
from .debugger.debugger import program_ids_from_grid
from . import language
from . import testing
@@ -43,7 +42,6 @@ __all__ = [
"runtime",
"TensorWrapper",
"testing",
"program_ids_from_grid",
]

View File

@@ -11,9 +11,6 @@ from .memory_map import MemoryMap
from .tl_lang import (TritonLangProxy, WrappedTensor, _primitive_to_tensor,
debugger_constexpr)
# import triton
torch = torch_wrapper.torch
tl_method_backup = {}

View File

@@ -2,7 +2,7 @@ from __future__ import annotations
import dataclasses
from . import torch_wrapper
from triton.interpreter import torch_wrapper
torch = torch_wrapper.torch

View File

@@ -529,7 +529,7 @@ def jit(
def decorator(fn: T) -> JITFunction[T]:
assert callable(fn)
if interpret:
from ..debugger.debugger import GridSelector
from ..interpreter.interpreter import GridSelector
return GridSelector(fn)
else:
return JITFunction(