diffsl 0.11.0

A compiler for a domain-specific language for ordinary differential equations (ODE).
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72

// RUN: %clang++ -std=c++11 -fno-exceptions -ffast-math -O2 %s -S -emit-llvm -o - | %opt - %OPloadEnzyme %enzyme -S | %lli -
// RUN: %clang++ -std=c++11 -fno-exceptions -ffast-math -O3 %s -S -emit-llvm -o - | %opt - %OPloadEnzyme %enzyme -S | %lli -
// RUN: %clang++ -std=c++11 -fno-exceptions -ffast-math -O2 %s -S -emit-llvm -o - | %opt - %OPloadEnzyme %enzyme -enzyme-inline=1 -S | %lli -
// RUN: %clang++ -std=c++11 -fno-exceptions -ffast-math -O3 %s -S -emit-llvm -o - | %opt - %OPloadEnzyme %enzyme -enzyme-inline=1 -S | %lli -

#include <stdio.h>
#include "../test_utils.h"

void __enzyme_autodiff(...);

template <typename... Args>
double __enzyme_autodiff(void *, Args...);

struct Object
{
    virtual double eval(double v) = 0;
    double val;
};

struct Object1 : Object
{
    double eval(double v)
    {
        return val * v;
    }
};

struct Object2 : Object
{
    double eval(double v)
    {
        return val + v;
    }
};

double eval(Object &o, double v)
{
    return o.eval(v);
}

double deval(Object &o, Object &d_o, double v)
{
    return __enzyme_autodiff((void *)(eval),
                      enzyme_dup, &o, &d_o,
                      v);
}

template<typename T>
T __enzyme_virtualreverse(T);

int main()
{
    double v = 10;
    Object *o = new Object1();
    o->val = 2;
    Object *d_o = new Object1();
    *((void**)d_o) = __enzyme_virtualreverse(*(void**)d_o);
    d_o->val = 0;
    double res;
    res = deval(*o, *d_o, v);
    printf("res=%f\n", res);
    APPROX_EQ(res, 2.0, 1e-7);
    
    o = new Object2();
    o->val = 2;
    d_o = new Object2();
    *((void**)d_o) = __enzyme_virtualreverse(*(void**)d_o);
    d_o->val = 0;
    res = deval(*o, *d_o, v);
    printf("res=%f\n", res);
    APPROX_EQ(res, 1.0, 1e-7);
}