[Fix] minor bugs

Author: YWHyuk

PyTorchSimFrontend/extension_config.py

@@ -67,7 +67,7 @@ def __getattr__(name):
             "multi_tile_conv",
             "subtile"
         }
-        if opt_level == "all" or opt_level is "none":
+        if opt_level == "all" or opt_level == "none":
             pass
         elif isinstance(opt_level, list):
             # Check if provided list contains only valid options

PyTorchSimFrontend/mlir/mlir_codegen_backend.py

@@ -1381,12 +1381,12 @@ def _index_expr(self, tile_desc, renamed_expression, index, base_vector_index):
         dim_list = []
         for idx in range(len(tile_size)):
             # Prepare initial values
-            offset = tile_desc.vlane_stride #* strides[idx]
-            outer_sz = tile_size[idx] // tile_desc.vlane_stride
+            offset = tile_desc.vmap.vlane_stride #* strides[idx]
+            outer_sz = tile_size[idx] // tile_desc.vmap.vlane_stride
             with self.override_buffer_cse(buffer=self.const_buffer, cse=self.const_cse):
                 div_coeff = self.get_const_cse(strides[idx], "index")
                 mod_coeff = self.get_const_cse(tile_size[idx], "index")
-                vlane_stride_coeff = self.get_const_cse(tile_desc.vlane_stride, "index")
+                vlane_stride_coeff = self.get_const_cse(tile_desc.vmap.vlane_stride, "index")
                 vlane_outer_coeff = self.get_const_cse(outer_sz, "index")
                 nr_vector_lane = self.get_const_cse(self.vector_lane, "index")
                 vlane_coeff = self.get_const_cse(0, "i64")

Scheduler/scheduler.py

@@ -179,7 +179,7 @@ def setup_device():
         )
 
         torch.utils.rename_privateuse1_backend("npu")
-        torch._register_device_module("extension_device", module)
+        torch._register_device_module("npu", module)
         from torch._inductor.codegen.common import (
             get_scheduling_for_device,
             get_wrapper_codegen_for_device,