Added Astaroth 2.0
This commit is contained in:
60
scripts/gen_rk3_threadblockconf.c
Normal file
60
scripts/gen_rk3_threadblockconf.c
Normal file
@@ -0,0 +1,60 @@
|
||||
/*
|
||||
Copyright (C) 2014-2018, Johannes Pekkilae, Miikka Vaeisalae.
|
||||
|
||||
This file is part of Astaroth.
|
||||
|
||||
Astaroth is free software: you can redistribute it and/or modify
|
||||
it under the terms of the GNU General Public License as published by
|
||||
the Free Software Foundation, either version 3 of the License, or
|
||||
(at your option) any later version.
|
||||
|
||||
Astaroth is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
GNU General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU General Public License
|
||||
along with Astaroth. If not, see <http://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
|
||||
/**
|
||||
* @file
|
||||
* \brief Generates a threadblock config file for RK3 using the given parameters.
|
||||
*
|
||||
*
|
||||
*/
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
|
||||
#include <assert.h>
|
||||
|
||||
const char* rk3_threadblockconf_path = "rk3_threadblock.conf";
|
||||
|
||||
int
|
||||
write_to_file(int threads_x, int threads_y, int threads_z, int elems_per_thread, int launch_bound)
|
||||
{
|
||||
FILE* fp;
|
||||
fp = fopen(rk3_threadblockconf_path, "w");
|
||||
|
||||
if (fp != NULL) {
|
||||
fprintf(fp, "#define RK_THREADS_X (%d)\n", threads_x);
|
||||
fprintf(fp, "#define RK_THREADS_Y (%d)\n", threads_y);
|
||||
fprintf(fp, "#define RK_THREADS_Z (%d)\n", threads_z);
|
||||
fprintf(fp, "#define RK_ELEMS_PER_THREAD (%d)\n", elems_per_thread);
|
||||
fprintf(fp, "#define RK_LAUNCH_BOUND_MIN_BLOCKS (%d)\n", launch_bound);
|
||||
fclose(fp);
|
||||
return EXIT_SUCCESS;
|
||||
}
|
||||
return EXIT_FAILURE;
|
||||
}
|
||||
|
||||
// Takes arguments and writes them into a file
|
||||
// RK_THREADS_X, RK_THREADS_Y, RK_THREADS_Z, RK_ELEMS_PER_THREAD, RK_LAUNCH_BOUND_MIN_BLOCKS
|
||||
int
|
||||
main(int argc, char* argv[])
|
||||
{
|
||||
assert(argc == 6);
|
||||
|
||||
return write_to_file(atoi(argv[1]), atoi(argv[2]),atoi(argv[3]), atoi(argv[4]), atoi(argv[5]));
|
||||
}
|
Reference in New Issue
Block a user