[Fix] minor bugs

YWHyuk · YWHyuk · commit 8ac4c603b95a · 2025-12-05T13:12:45.000Z
diff --git a/PyTorchSimFrontend/extension_config.py b/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
diff --git a/Scheduler/scheduler.py b/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,

Original file line number	Diff line number	Diff line change
`@@ -67,7 +67,7 @@ def __getattr__(name):`
`67`	`67`	`"multi_tile_conv",`
`68`	`68`	`"subtile"`
`69`	`69`	`}`
`70`		`- if opt_level == "all" or opt_level is "none":`
	`70`	`+ if opt_level == "all" or opt_level == "none":`
`71`	`71`	`pass`
`72`	`72`	`elif isinstance(opt_level, list):`
`73`	`73`	`# Check if provided list contains only valid options`
Original file line number	Diff line number	Diff line change
`@@ -179,7 +179,7 @@ def setup_device():`
`179`	`179`	`)`
`180`	`180`
`181`	`181`	`torch.utils.rename_privateuse1_backend("npu")`
`182`		`- torch._register_device_module("extension_device", module)`
	`182`	`+ torch._register_device_module("npu", module)`
`183`	`183`	`from torch._inductor.codegen.common import (`
`184`	`184`	`get_scheduling_for_device,`
`185`	`185`	`get_wrapper_codegen_for_device,`