pseXperiments/icicle
1
0
Fork 0
mirror of https://github.com/pseXperiments/icicle.git synced 2026-01-09 21:17:56 -05:00
Code Issues Packages Projects Releases Wiki Activity

fixed on-device memory issue

Browse Source
This commit is contained in:
Stas Polonsky
2024-02-16 19:43:58 +00:00
parent 319358427f
commit 6681be549a
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
examples/c++/multi-gpu-poseidon/example.cu
View File

@@ -37,7 +37,7 @@ void threadPoseidon(device_context::DeviceContext ctx, unsigned size_partition,
false, // input_is_a_state
false, // aligned
false, // loop_state
true, // is_async
false, // is_async
};
cudaError_t err = poseidon_hash<scalar_t, size_col+1>(layers, column_hashes, (size_t) size_partition, *constants, column_config);
checkCudaError(err);
@@ -102,6 +102,11 @@ int main() {
PoseidonConstants<scalar_t> column_constants0, column_constants1;
init_optimized_poseidon_constants<scalar_t>(size_col, ctx0, &column_constants0);
cudaError_t err_result = CHK_STICKY(cudaSetDevice(ctx1.device_id));
if (err_result != cudaSuccess) {
std::cerr << "CUDA error: " << cudaGetErrorString(err_result) << std::endl;
return;
}
init_optimized_poseidon_constants<scalar_t>(size_col, ctx1, &column_constants1);
std::cout << "Parallel execution of Poseidon threads" << std::endl;