mirror of
https://github.com/ROCm/ROCm.git
synced 2026-04-05 03:01:17 -04:00
[FRONTEND] cleaned up symbol names (#1782)
This commit is contained in:
@@ -265,7 +265,7 @@ setup(
|
||||
"triton/_C",
|
||||
"triton/common",
|
||||
"triton/compiler",
|
||||
"triton/debugger",
|
||||
"triton/interpreter",
|
||||
"triton/language",
|
||||
"triton/language/extra",
|
||||
"triton/ops",
|
||||
|
||||
@@ -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():
|
||||
@@ -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",
|
||||
]
|
||||
|
||||
|
||||
|
||||
@@ -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 = {}
|
||||
|
||||
@@ -2,7 +2,7 @@ from __future__ import annotations
|
||||
|
||||
import dataclasses
|
||||
|
||||
from . import torch_wrapper
|
||||
from triton.interpreter import torch_wrapper
|
||||
|
||||
torch = torch_wrapper.torch
|
||||
|
||||
@@ -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(
|
||||
|
||||
Reference in New Issue
Block a user