Method compiler_ir

imperative/python/megengine/xla/compile.py:577–587

(self, dialect: Optional[str] = None)

            raise ValueError(f"unknown dialect: {dialect}")

    def compiler_ir(self, dialect: Optional[str] = None) -> Any:
        if dialect is None:
            dialect = "stablehlo"
        if dialect == "mhlo":
            return self.mhlo()
        elif dialect == "stablehlo":
            return self.stablehlo()
        elif dialect == "hlo":
            return self.hlo()
        else:
            raise ValueError(f"unknown dialect: {dialect}")


class MeshComputation(XlaLowering):