Skip to content

Commit

Permalink
let 3.8 compatiable (#7210)
Browse files Browse the repository at this point in the history
  • Loading branch information
zpcore committed Jun 6, 2024
1 parent 0e1f765 commit 8748a2f
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions torch_xla/experimental/custom_kernel.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,15 +8,15 @@
import torch_xla.core.xla_model as xm
import torch_xla.distributed.spmd as xs

from typing import Any, List, Callable
from typing import Any, List, Callable, Optional
from torch.library import impl
from torch_xla.core.xla_model import XLA_LIB

_XLA_USE_BF16 = os.environ.get("XLA_USE_BF16", "0") == "1"


def _extract_backend_config(
module: "jaxlib.mlir._mlir_libs._mlir.ir.Module") -> str | None:
module: "jaxlib.mlir._mlir_libs._mlir.ir.Module") -> Optional[str]:
"""
This algorithm intends to extract the backend config from the compiler IR like the following,
and it is not designed to traverse any generic MLIR module.
Expand Down

0 comments on commit 8748a2f

Please sign in to comment.