(git:e7e05ae)
dbm_multiply_cpu.h
Go to the documentation of this file.
1 /*----------------------------------------------------------------------------*/
2 /* CP2K: A general program to perform molecular dynamics simulations */
3 /* Copyright 2000-2024 CP2K developers group <https://cp2k.org> */
4 /* */
5 /* SPDX-License-Identifier: BSD-3-Clause */
6 /*----------------------------------------------------------------------------*/
7 #ifndef DBM_MULTIPLY_CPU_H
8 #define DBM_MULTIPLY_CPU_H
9 
10 #include <stdbool.h>
11 
12 #include "dbm_multiply_internal.h"
13 #include "dbm_shard.h"
14 
15 /*******************************************************************************
16  * \brief Internal routine for executing the tasks in given batch on the CPU.
17  * \author Ole Schuett
18  ******************************************************************************/
19 void dbm_multiply_cpu_process_batch(const int ntasks, dbm_task_t batch[ntasks],
20  const double alpha,
21  const dbm_pack_t *pack_a,
22  const dbm_pack_t *pack_b,
23  dbm_shard_t *shard_c);
24 
25 #endif
26 
27 // EOF
void dbm_multiply_cpu_process_batch(const int ntasks, dbm_task_t batch[ntasks], const double alpha, const dbm_pack_t *pack_a, const dbm_pack_t *pack_b, dbm_shard_t *shard_c)
Internal routine for executing the tasks in given batch on the CPU.
Internal struct for storing a pack - essentially a shard for MPI.
Internal struct for storing a matrix shard.
Definition: dbm_shard.h:30
Internal struct for storing a task, ie. a single block multiplication.