[Wave] Rename wave.* -> wave_lang.* (#47)

This PR renames the top level wave directory
to wave_lang to avoid conflicts with Python's
native wave package.

Signed-off-by: Harsh Menon <[email protected]>

Author: harsh-nod

iree/turbine/kernel/wave/codegen/emitter.py

@@ -14,7 +14,7 @@
 import torch.fx as fx
 
 from iree.turbine.kernel.lang.global_symbols import *
-from wave.aot.support.ir_utils import (
+from wave_lang.aot.support.ir_utils import (
     _is_float_type,
     _is_integer_like_type,
 )

iree/turbine/kernel/wave/codegen/handlers.py

@@ -37,7 +37,7 @@
     scf_d,
     vector_d,
 )
-from wave.aot.support.ir_utils import (
+from wave_lang.aot.support.ir_utils import (
     _is_float_type,
     _is_integer_like_type,
     _is_signed_or_signless_type,

iree/turbine/runtime/op_reg/impl_helper.py

@@ -36,7 +36,7 @@ def generate(kb: KernelBuilder):
     Value,
 )
 
-from wave.transforms.merger import Merger
+from wave_lang.transforms.merger import Merger
 
 from .base import (
     KernelBuilder,

lit_tests/kernel/wave/sharktank_integration.py

@@ -1,14 +1,13 @@
 # RUN: python %s | FileCheck %s
 
 import textwrap
-import wave.aot as aot
 from typing import Optional
-from wave.transforms.merger import Merger
 
 import torch
 from jinja2 import BaseLoader, Environment
 
 import iree.turbine.kernel.wave as tkw
+import wave_lang.aot as aot
 from iree.compiler.ir import (
     Context,
     MLIRError,
@@ -44,6 +43,7 @@
 from iree.turbine.runtime.op_reg.impl_helper import (
     call_function,
 )
+from wave_lang.transforms.merger import Merger
 
 _JINJA2_ENVIRONMENT: Optional[Environment] = None
 

setup.py

@@ -76,8 +76,8 @@ def load_version_info(version_file):
     include=[
         "iree.turbine",
         "iree.turbine.*",
-        "wave",
-        "wave.*",
+        "wave_lang",
+        "wave_lang.*",
     ],
 )
 

tests/tools/interpreter_test.py

@@ -1,4 +1,4 @@
-from wave.tools.interpreter import Interpreter
+from wave_lang.tools.interpreter import Interpreter
 import iree.turbine.kernel.lang as tkl
 import iree.turbine.kernel.wave as tkw
 from iree.turbine.kernel.lang.global_symbols import *