cg_blas.c
No OneTemporary
Actions

Subscribers

None

File Metadata

	#include <stdio.h>
	#include <stdlib.h>
	#include <math.h>
	#include <stdarg.h>
	#include <stddef.h>

	#include "blas.h"
	#include "cg.h"
	#include "parameters.h"



	/*
	main function
	*/
	int main ( int argc, char **argv ) {

	double * A;
	double * At;
	double * b;
	double * x;
	double * x0;
	//double * x_check;

	int n,m;

	n = 2;
	m = n;

	A = (double) malloc(mn* sizeof(double));
	At = (double) malloc(mn* sizeof(double));
	b = (double) malloc(n sizeof(double));
	x = (double) malloc(n sizeof(double));
	x0 = (double) malloc(n sizeof(double));
	//x_check = (double) malloc(n sizeof(double));

	// {{ 4, 1},{ 1, 3}}

	A[id(n,0,0)] = 4.;
	A[id(n,0,1)] = 1.;
	A[id(n,1,0)] = 1.;
	A[id(n,1,1)] = 3.;

	cblas_dtra(m, n, A, At);

	// vec B = { 1, 2 };
	b[0] = 1.;
	b[1] = 2.;

	x0[0] = 2;
	x0[1] = 1;

	// print_mat( "matrice A:\n", A,m,n );
	// print_mat( "matrice A transpose:\n", At,m,n );
	// print_mat( "vecteur b: \n", b,1,n );

	cgsolver( A, b, x, m, n );

	printf("Solves AX = B\n");
	print_mat( "\nX:", x,1,n );

	cblas_dgemv (1, 0, m, n, 1., A, 1., x, 1, 1., x, 1.);

	print_mat( "\nCheck AX:\n", x0,1, n);
	/*
	free(A);
	free(b);
	free(x);
	free(x0);
	free(x_check);
	*/

	return 0;
	}