Unverified Commit b1922518 authored by Yichen Yan's avatar Yichen Yan Committed by GitHub
Browse files

[Minor] Remove from __future__ import annotations for python 3.8 (#1273)

parent 220c3236
"""Wrapping Layouts.""" """Wrapping Layouts."""
# pylint: disable=invalid-name, unsupported-binary-operation # pylint: disable=invalid-name, unsupported-binary-operation
from __future__ import annotations from __future__ import annotations
import tvm import tvm
from tvm.tir import Buffer, BufferLoad, BufferRegion from tvm.tir import Buffer, BufferLoad, BufferRegion
from tilelang import _ffi_api from tilelang import _ffi_api
......
from __future__ import annotations from __future__ import annotations
from tvm import tir from tvm import tir
from tilelang.utils import is_local, is_fragment, is_shared from tilelang.utils import is_local, is_fragment, is_shared
from tilelang.primitives.gemm.base import GemmWarpPolicy from tilelang.primitives.gemm.base import GemmWarpPolicy
......
"""The profiler and convert to torch utils""" """The profiler and convert to torch utils"""
from __future__ import annotations from __future__ import annotations
from typing import Callable, Any, Literal from typing import Callable, Any, Literal
from functools import partial from functools import partial
import torch import torch
......
# Copyright (c) Microsoft Corporation. # Copyright (c) Microsoft Corporation.
# Licensed under the MIT License. # Licensed under the MIT License.
from __future__ import annotations
from typing import Literal from typing import Literal
decode_i4_to_f16 = """ decode_i4_to_f16 = """
......
from __future__ import annotations
from typing import Literal from typing import Literal
# Implementation asm for fp4 to bf16, using twiddling # Implementation asm for fp4 to bf16, using twiddling
......
from __future__ import annotations
from tvm.tir import (BufferStore, For, AttrStmt, ForKind, Var, PrimFunc, BufferLoad, Buffer, IntImm) from tvm.tir import (BufferStore, For, AttrStmt, ForKind, Var, PrimFunc, BufferLoad, Buffer, IntImm)
from tvm.tir.stmt_functor import ir_transform, post_order_visit from tvm.tir.stmt_functor import ir_transform, post_order_visit
from tvm.tir.transform import prim_func_pass from tvm.tir.transform import prim_func_pass
......
from __future__ import annotations
"""The profiler and convert to torch utils""" """The profiler and convert to torch utils"""
from enum import Enum from enum import Enum
import torch import torch
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment