fix import

jiqing-feng · jiqing-feng · commit bde878a0c855 · 2024-11-21T12:43:17.000Z
diff --git a/bitsandbytes/nn/__init__.py b/bitsandbytes/nn/__init__.py
@@ -17,7 +17,7 @@
 )
 
 # CPU and XPU backend do not need triton, and XPU so not support triton for now.
-if "xpu" not in backends.keys() or ("cpu" in backends.keys() and len(backends.keys()) == 1):
+if "xpu" not in backends.keys() and len(backends.keys()) > 1:
     from .triton_based_modules import (
         StandardLinear,
         SwitchBackLinear,

Original file line number	Diff line number	Diff line change
`@@ -17,7 +17,7 @@`
`17`	`17`	`)`
`18`	`18`
`19`	`19`	`# CPU and XPU backend do not need triton, and XPU so not support triton for now.`
`20`		`-if "xpu" not in backends.keys() or ("cpu" in backends.keys() and len(backends.keys()) == 1):`
	`20`	`+if "xpu" not in backends.keys() and len(backends.keys()) > 1:`
`21`	`21`	`from .triton_based_modules import (`
`22`	`22`	`StandardLinear,`
`23`	`23`	`SwitchBackLinear,`