Fix circular import in aitemplate (#951)

Summary:
Pull Request resolved: #951

Fixes import cycle: P852820857

Reviewed By: cxxxs, chenyang78

Differential Revision: D50283535

fbshipit-source-id: e9e7fbc054b5312d1c9a073a0554d0c3a3788cc0

python/aitemplate/compiler/transform/fuse_expand_bmm.py

@@ -26,7 +26,6 @@
 from typing import List
 
 from aitemplate.compiler.base import Operator, Tensor
-from aitemplate.compiler.ops.tensor.expand import ExpandDimensionType
 from aitemplate.compiler.tensor_accessor import TensorAccessor
 from aitemplate.compiler.transform.toposort import toposort
 from aitemplate.compiler.transform.transform_utils import (
@@ -39,6 +38,10 @@ def _can_fuse(expand_op: Operator, bmm_op: Operator) -> bool:
     """
     determine if expand_op and bmm_op can be fused
     """
+    from aitemplate.compiler.ops.tensor.expand import (  # inner import to break circular import
+        ExpandDimensionType,
+    )
+
     expand_output = expand_op._attrs["outputs"][0]
     if expand_output._attrs["is_output"]:
         return False