kaushikcfd · November 19, 2021 13:46
diff --git a/axpy.mlir b/axpy.mlir
 func @saxpy(%a : f32, %x : memref<?xf32>, %y : memref<?xf32>) {
  %c0 = constant 0: index
  %n = dim %x, %c0 : memref<?xf32>

  affine.for %i = 0 to %n {
    %xi = affine.load %x[%i] : memref<?xf32>
    %axi =  mulf %a, %xi : f32
    %yi = affine.load %y[%i] : memref<?xf32>
    %axpyi = addf %yi, %axi : f32
    affine.store %axpyi, %y[%i] : memref<?xf32>
  }
  return
 }
diff --git a/call_axpy.c b/call_axpy.c
 #include <stdio.h>
 struct OneDMemrefF32 {
  float* ptrToData;
  float* alignedPtrToData;
  float offset;
  long shape[1];
  long stride[1];
 };

 #define N 4
 #define REAL float
 #define AXPY _mlir_ciface_saxpy
 #define MEMREF OneDMemrefF32

 extern void _mlir_ciface_saxpy(float, struct OneDMemrefF32*, struct OneDMemrefF32*);



 int main() {
  REAL x[N], y[N], alpha = 2.0;
  struct MEMREF xMemref = {x, x, 0, {N}, {1}};
  struct MEMREF yMemref = {y, y, 0, {N}, {1}};

  for (int i=0; i<N; ++i) {
    x[i] = i*i;
    y[i] = i;
  }

  printf("x: {");
  for (int i=0; i<N-1; ++i)
    printf("%3.1f, ", x[i]);
  printf("%3.1f}\n", x[N-1]);

  printf("y: {");
  for (int i=0; i<N-1; ++i)
    printf("%3.1f, ", y[i]);
  printf("%3.1f}\n", y[N-1]);


  AXPY(alpha, &xMemref, &yMemref);

  printf("%3.2fx + y: {", alpha);
  for (int i=0; i<N-1; ++i)
    printf("%3.1f, ", y[i]);
  printf("%3.1f}\n", y[N-1]);
 }
diff --git a/MLIR-From-C.md b/MLIR-From-C.md
	func @saxpy(%a : f32, %x : memref<?xf32>, %y : memref<?xf32>) {
	%c0 = constant 0: index
	%n = dim %x, %c0 : memref<?xf32>

	affine.for %i = 0 to %n {
	%xi = affine.load %x[%i] : memref<?xf32>
	%axi = mulf %a, %xi : f32
	%yi = affine.load %y[%i] : memref<?xf32>
	%axpyi = addf %yi, %axi : f32
	affine.store %axpyi, %y[%i] : memref<?xf32>
	}
	return
	}
	#include <stdio.h>
	struct OneDMemrefF32 {
	float* ptrToData;
	float* alignedPtrToData;
	float offset;
	long shape[1];
	long stride[1];
	};

	#define N 4
	#define REAL float
	#define AXPY _mlir_ciface_saxpy
	#define MEMREF OneDMemrefF32

	extern void _mlir_ciface_saxpy(float, struct OneDMemrefF32, struct OneDMemrefF32);



	int main() {
	REAL x[N], y[N], alpha = 2.0;
	struct MEMREF xMemref = {x, x, 0, {N}, {1}};
	struct MEMREF yMemref = {y, y, 0, {N}, {1}};

	for (int i=0; i<N; ++i) {
	x[i] = i*i;
	y[i] = i;
	}

	printf("x: {");
	for (int i=0; i<N-1; ++i)
	printf("%3.1f, ", x[i]);
	printf("%3.1f}\n", x[N-1]);

	printf("y: {");
	for (int i=0; i<N-1; ++i)
	printf("%3.1f, ", y[i]);
	printf("%3.1f}\n", y[N-1]);


	AXPY(alpha, &xMemref, &yMemref);

	printf("%3.2fx + y: {", alpha);
	for (int i=0; i<N-1; ++i)
	printf("%3.1f, ", y[i]);
	printf("%3.1f}\n", y[N-1]);
	}