mirror of
https://github.com/TREX-CoE/Sherman-Morrison.git
synced 2024-12-26 06:15:08 +01:00
63 lines
2.7 KiB
C
63 lines
2.7 KiB
C
#pragma once
|
|
|
|
#include <math.h>
|
|
#include <stdio.h>
|
|
#include <stdint.h>
|
|
#include <assert.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <hdf5/serial/hdf5.h>
|
|
|
|
#include "kernels.h"
|
|
|
|
typedef struct Error {
|
|
uint32_t rc;
|
|
uint64_t error;
|
|
} Error;
|
|
|
|
#ifdef USE_OMP_OFFLOAD_CUDA
|
|
cublasHandle_t init_cublas();
|
|
cusolverDnHandle_t init_cusolver();
|
|
#endif
|
|
|
|
void copy(double* Slater_invT_copy, uint64_t Lds, double* tmp, uint64_t Dim);
|
|
void update(double* slaterT,double* upds, uint64_t* ui, uint64_t nupds,uint64_t Dim, u_int64_t Lds);
|
|
void convert(double* upds, uint64_t nupds, uint64_t* ui, double* slaterT, uint64_t Dim, u_int64_t Lds);
|
|
void transpose(double* a, uint16_t lda, double *b, uint16_t ldb, uint16_t m, uint16_t n);
|
|
double get_determinant(uint32_t cycle, hid_t file_id);
|
|
double* get_slater_inv(uint32_t cycle, hid_t file_id, uint64_t Dim, u_int64_t Lds);
|
|
double* get_slater(uint32_t cycle, hid_t file_id, uint64_t Dim, u_int64_t Lds);
|
|
double* get_upds(uint32_t cycle, hid_t file_id, uint64_t nupds, u_int64_t Lds);
|
|
uint64_t* get_upd_idcs(uint32_t cycle, hid_t file_id, uint64_t nupds);
|
|
uint64_t get_dim(uint32_t cycle, hid_t file_id);
|
|
uint64_t get_nupdates(uint32_t cycle, hid_t file_id);
|
|
|
|
//void matmul(double *a, double *b, double *prod, const uint64_t Lds, const uint64_t Dim);
|
|
void residual(double *a, double *res, const uint64_t Dim);
|
|
double frobenius_norm2(double *A, const uint64_t Lds, const uint64_t Dim);
|
|
double frobenius_norm(double *A, const uint64_t Lds, const uint64_t Dim);
|
|
double max_norm(double *A, const uint64_t Lds, const uint64_t Dim);
|
|
double condition_number(double *A, double *Ainv, const uint64_t Lds, const uint64_t Dim);
|
|
void read_uint(hid_t file_id, const char *key, uint64_t *data);
|
|
void read_double(hid_t file_id, const char *key, double *data);
|
|
|
|
static __inline__ uint64_t rdtsc(void) {
|
|
unsigned hi, lo;
|
|
__asm__ __volatile__("rdtsc" : "=a"(lo), "=d"(hi));
|
|
return ((unsigned long long)lo) | (((unsigned long long)hi) << 32);
|
|
}
|
|
|
|
void update_slater_matrix(const uint64_t Lds, const uint64_t Dim,
|
|
const uint64_t N_updates, const double *Updates,
|
|
const uint64_t *Updates_index, double *Slater);
|
|
|
|
uint32_t check_error(const uint64_t Lds, const uint64_t Dim, double *Slater_invT,
|
|
double *Slater, const double tolerance);
|
|
|
|
int32_t check_error_better(const double max, const double tolerance);
|
|
|
|
uint32_t test_kernel(char *version, const uint64_t Lds, const uint64_t Dim,
|
|
const uint64_t N_updates, const double *Updates,
|
|
const uint64_t *Updates_index, const double breakdown, const double tolerance,
|
|
double *Slater, double *Slater_inv, double *determinant);
|