Skip to content

Commit 382043e

Browse files
committed
Bump Enzyme abi
1 parent 5c141f5 commit 382043e

File tree

2 files changed

+11
-1
lines changed

2 files changed

+11
-1
lines changed

compiler/rustc_codegen_llvm/src/llvm/ffi.rs

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,8 @@
11
#![allow(non_camel_case_types)]
22
#![allow(non_upper_case_globals)]
33

4+
5+
use std::ptr;
46
use rustc_codegen_ssa::coverageinfo::map as coverage_map;
57
use rustc_middle::middle::autodiff_attrs::DiffActivity;
68

@@ -1062,6 +1064,8 @@ pub(crate) unsafe fn enzyme_rust_forward_diff(
10621064
EnzymeCreateForwardDiff(
10631065
logic_ref, // Logic
10641066
fnc,
1067+
ptr::null(),
1068+
ptr::null(),
10651069
ret_activity, // LLVM function, return type
10661070
input_activity.as_ptr(),
10671071
input_activity.len(), // constant arguments
@@ -1124,6 +1128,8 @@ pub(crate) unsafe fn enzyme_rust_reverse_diff(
11241128
EnzymeCreatePrimalAndGradient(
11251129
logic_ref, // Logic
11261130
fnc,
1131+
ptr::null(),
1132+
ptr::null(),
11271133
ret_activity, // LLVM function, return type
11281134
input_activity.as_ptr(),
11291135
input_activity.len(), // constant arguments
@@ -2819,6 +2825,8 @@ extern "C" {
28192825
fn EnzymeCreatePrimalAndGradient<'a>(
28202826
arg1: EnzymeLogicRef,
28212827
todiff: &'a Value,
2828+
builderCtx: *const u8, // &'a Builder<'_>,
2829+
callerCtx: *const u8,// &'a Value,
28222830
retType: CDIFFE_TYPE,
28232831
constant_args: *const CDIFFE_TYPE,
28242832
constant_args_size: size_t,
@@ -2842,6 +2850,8 @@ extern "C" {
28422850
fn EnzymeCreateForwardDiff<'a>(
28432851
arg1: EnzymeLogicRef,
28442852
todiff: &'a Value,
2853+
builderCtx: *const u8,// &'a Builder<'_>,
2854+
callerCtx: *const u8,// &'a Value,
28452855
retType: CDIFFE_TYPE,
28462856
constant_args: *const CDIFFE_TYPE,
28472857
constant_args_size: size_t,

src/tools/enzyme

Submodule enzyme updated 82 files

0 commit comments

Comments
 (0)