Add multisource

Author: wsmoses

6_multisource/Makefile

@@ -0,0 +1,13 @@
+all: lib.exe
+
+clean:
+	rm -f *.o *.ll *.exe
+
+%.o: %.c
+	../dockerscript.sh clang-12 -c -fuse-ld=lld -flto /host/$^ -O2 -ffast-math -o /host/$@
+
+lib.exe: myblas.o multisource.o
+	../dockerscript.sh clang-12 -fuse-ld=lld -flto /host/myblas.o /host/multisource.o -O2 -ffast-math -o /host/$@ -Wl,-mllvm=-load=/Enzyme/enzyme/build/Enzyme/ClangEnzyme-12.so
+
+run-%: %.exe
+	../dockerscript.sh /host/$^ 3

6_multisource/multisource.c

@@ -0,0 +1,48 @@
+#include <stdio.h>
+#include <stdlib.h>
+#include <math.h>
+#include <assert.h>
+#include "myblas.h"
+
+double dotabs(struct complex* alpha, struct complex* beta, int n) {
+  struct complex prod = myblas_cdot(alpha, beta, n);
+  return myblas_cabs(prod);
+}
+
+void __enzyme_autodiff(void*, ...);
+int enzyme_const, enzyme_dup, enzyme_out;
+
+int main(int argc, char *argv[]) {
+  int n = 3;
+  if (argc > 1) {
+    n = atoi(argv[1]);
+  }
+
+
+  struct complex *A = (struct complex*)malloc(sizeof(struct complex) * n);
+  assert(A != 0);
+  for(int i=0; i<n; i++)
+   A[i] = (struct complex){(i+1), (i+2)};
+
+  struct complex *grad_A = (struct complex*)malloc(sizeof(struct complex) * n);
+  assert(grad_A != 0);
+  for(int i=0; i<n; i++)
+   grad_A[i] = (struct complex){0,0};
+
+
+
+  struct complex *B = (struct complex*)malloc(sizeof(struct complex) * n);
+  assert(B != 0);
+  for(int i=0; i<n; i++)
+   B[i] = (struct complex){-3-i, 2*i};
+
+  struct complex *grad_B = (struct complex*)malloc(sizeof(struct complex) * n);
+  assert(grad_B != 0);
+  for(int i=0; i<n; i++)
+   grad_B[i] = (struct complex){0,0};
+
+  __enzyme_autodiff((void*)dotabs, A, grad_A, B, grad_B, n);
+  printf("Gradient dotabs(A)[0] = %f\n", grad_A[0].r);
+
+  return 0;
+}

6_multisource/myblas.c

@@ -0,0 +1,14 @@
+#include "myblas.h"
+
+struct complex myblas_cdot(struct complex* x, struct complex *y, int n) {
+  struct complex sum = { 0, 0 };
+  for (int i=0; i<n; i++) {
+    sum.r += x[i].r * y[i].r - x[i].i * y[i].i;
+    sum.i += x[i].r * y[i].i + x[i].i * y[i].r;
+  }
+  return sum;
+}
+
+double myblas_cabs(struct complex x) {
+    return x.r * x.r + x.i * x.i;
+}

6_multisource/myblas.h

@@ -0,0 +1,8 @@
+struct complex {
+    double r;
+    double i;
+};
+
+struct complex myblas_cdot(struct complex* x, struct complex *y, int n);
+
+double myblas_cabs(struct complex x);

docker/Dockerfile

@@ -5,7 +5,7 @@ ENV DEBIAN_FRONTEND=noninteractive
 RUN apt-get -y update && apt-get install -y --no-install-recommends curl gnupg lsb-core software-properties-common \
     && curl -fsSL https://apt.llvm.org/llvm-snapshot.gpg.key|apt-key add - \
     && apt-add-repository "deb http://apt.llvm.org/`lsb_release -c | cut -f2`/ llvm-toolchain-`lsb_release -c | cut -f2`-12 main" \
-    && apt-get install -y --no-install-recommends autoconf cmake ninja-build gcc g++ libtool gfortran llvm-12-dev clang-12 libopenmpi-dev openmpi-bin git \
+    && apt-get install -y --no-install-recommends autoconf cmake ninja-build gcc g++ libtool gfortran llvm-12-dev lld-12 clang-12 libopenmpi-dev openmpi-bin git \
     && apt-get autoremove -y --purge \
     && apt-get clean -y \
     && rm -rf /var/lib/apt/lists/*
@@ -19,6 +19,7 @@ RUN git clone https://github.com/wsmoses/Enzyme.git \
 
 RUN update-alternatives --install /usr/bin/clang clang /usr/bin/clang-12 10 --slave /usr/bin/clang++ clang++ /usr/bin/clang++-12 \
     && update-alternatives --install /usr/bin/opt opt /usr/bin/opt-12 10 \
+    && update-alternatives --install /usr/bin/lld lld /usr/bin/lld-12 10 \
     && update-alternatives --install /usr/bin/llvm-symbolizer llvm-symbolizer /usr/bin/llvm-symbolizer-12 10
 
-ENV DEBIAN_FRONTEND=
+ENV DEBIAN_FRONTEND=