From 8f765d3e39f778aa6526da94cd990ded88d4c611 Mon Sep 17 00:00:00 2001 From: Nazar Mokrynskyi Date: Sun, 6 Oct 2024 23:37:15 +0300 Subject: [PATCH] Fix small GPU support --- shared/subspace-proof-of-space-gpu/src/chacha.cuh | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/shared/subspace-proof-of-space-gpu/src/chacha.cuh b/shared/subspace-proof-of-space-gpu/src/chacha.cuh index 2c40ea2488..10bed1ce68 100644 --- a/shared/subspace-proof-of-space-gpu/src/chacha.cuh +++ b/shared/subspace-proof-of-space-gpu/src/chacha.cuh @@ -101,12 +101,13 @@ void chacha_mix(chacha_state& state, size_t counter) template __global__ __launch_bounds__(1024) -void generate_chacha_keystream(chacha_state* output, chacha_state state, +void generate_chacha_keystream(chacha_state* output, chacha_state _state, size_t output_block_count) { size_t tid = blockIdx.x * (size_t)blockDim.x + threadIdx.x; for (size_t i = tid; i < output_block_count; i += (size_t)gridDim.x * blockDim.x) { + chacha_state state = _state; chacha_mix(state, i); output[i] = state;