Skip to content

Commit 5689a40

Browse files
authored
Merge pull request #12 from pynbody/max-buffer-size
Make max_buffer_size as big as possible
2 parents fa45148 + cae75ef commit 5689a40

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

src/topsy/visualizer.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -79,8 +79,13 @@ def __init__(self, data_loader_class = loader.TestDataLoader, data_loader_args =
7979
def _setup_wgpu(self):
8080
self.adapter: wgpu.GPUAdapter = wgpu.gpu.request_adapter(power_preference="high-performance")
8181
if self.device is None:
82+
max_buffer_size = self.adapter.limits['max_buffer_size']
83+
# on some systems, this is 2^64 which can lead to overflows
84+
if max_buffer_size > 2**63:
85+
max_buffer_size = 2**63
8286
type(self).device: wgpu.GPUDevice = self.adapter.request_device(
83-
required_features=["TextureAdapterSpecificFormatFeatures", "float32-filterable"])
87+
required_features=["TextureAdapterSpecificFormatFeatures", "float32-filterable"],
88+
required_limits={"max_buffer_size": max_buffer_size})
8489
self.context: wgpu.GPUCanvasContext = self.canvas.get_context()
8590
self.canvas_format = self.context.get_preferred_format(self.adapter)
8691
if self.canvas_format.endswith("-srgb"):

0 commit comments

Comments
 (0)