(git:97501a3)
Loading...
Searching...
No Matches
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-2025 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 "
dbm_internal.h
"
11
#include "
dbm_shard.h
"
12
13
enum
dbm_multiply_cpu_options
{
14
DBM_MULTIPLY_BLAS_LIBRARY
= 1,
15
DBM_MULTIPLY_TASK_REORDER
= 2
16
};
17
18
/*******************************************************************************
19
* \brief Internal routine for executing the tasks in given batch on the CPU.
20
* \author Ole Schuett
21
******************************************************************************/
22
void
dbm_multiply_cpu_process_batch
(
int
ntasks,
const
dbm_task_t
batch[ntasks],
23
double
alpha,
const
dbm_pack_t
*pack_a,
24
const
dbm_pack_t
*pack_b,
25
dbm_shard_t
*shard_c,
int
options);
26
27
#endif
28
29
// EOF
dbm_internal.h
dbm_multiply_cpu_options
dbm_multiply_cpu_options
Definition
dbm_multiply_cpu.h:13
DBM_MULTIPLY_BLAS_LIBRARY
@ DBM_MULTIPLY_BLAS_LIBRARY
Definition
dbm_multiply_cpu.h:14
DBM_MULTIPLY_TASK_REORDER
@ DBM_MULTIPLY_TASK_REORDER
Definition
dbm_multiply_cpu.h:15
dbm_multiply_cpu_process_batch
void dbm_multiply_cpu_process_batch(int ntasks, const dbm_task_t batch[ntasks], double alpha, const dbm_pack_t *pack_a, const dbm_pack_t *pack_b, dbm_shard_t *shard_c, int options)
Internal routine for executing the tasks in given batch on the CPU.
Definition
dbm_multiply_cpu.c:68
dbm_shard.h
dbm_pack_t
Internal struct for storing a pack - essentially a shard for MPI.
Definition
dbm_internal.h:32
dbm_shard_t
Internal struct for storing a matrix shard.
Definition
dbm_shard.h:30
dbm_task_t
Internal struct for storing a task, ie. a single block multiplication.
Definition
dbm_internal.h:43
dbm
dbm_multiply_cpu.h
Generated by
1.9.8