update compute_type_is_set attr (#1623)

ckvermaAI · web-flow · commit 5e267f5fde01 · 2025-05-05T15:26:05.000-04:00
diff --git a/bitsandbytes/nn/modules.py b/bitsandbytes/nn/modules.py
@@ -447,7 +447,7 @@ def __init__(
         )
         # self.persistent_buffers = []  # TODO consider as way to save quant state
         self.compute_dtype = compute_dtype
-        self.compute_type_is_set = False
+        self.compute_type_is_set = False if compute_dtype is None else True
         self.ipex_linear_is_set = False
         self.quant_state = None
         self.quant_storage = quant_storage

Original file line number	Diff line number	Diff line change
`@@ -447,7 +447,7 @@ def __init__(`
`447`	`447`	`)`
`448`	`448`	`# self.persistent_buffers = [] # TODO consider as way to save quant state`
`449`	`449`	`self.compute_dtype = compute_dtype`
`450`		`- self.compute_type_is_set = False`
	`450`	`+ self.compute_type_is_set = False if compute_dtype is None else True`
`451`	`451`	`self.ipex_linear_is_set = False`
`452`	`452`	`self.quant_state = None`
`453`	`453`	`self.quant_storage = quant_storage`