MCPcopy Create free account
hub / github.com/bitsandbytes-foundation/bitsandbytes / cuda

Method cuda

bitsandbytes/nn/modules.py:753–754  ·  view source on GitHub ↗
(self, device: Optional[int | device | str] = None, non_blocking: bool = False)

Source from the content-addressed store, hash-verified

751 return self.to(device="cpu")
752
753 def cuda(self, device: Optional[int | device | str] = None, non_blocking: bool = False):
754 return self.to(device="cuda" if device is None else device, non_blocking=non_blocking)
755
756 def xpu(self, device: Optional[int | device | str] = None, non_blocking: bool = False):
757 return self.to(device="xpu" if device is None else device, non_blocking=non_blocking)

Callers

nothing calls this directly

Calls 1

toMethod · 0.95

Tested by

no test coverage detected