Skip to content

Commit d74c622

Browse files
authored
Merge pull request #55 from EnzymeAD/bumpabi
Bump Enzyme abi
2 parents 5c141f5 + d57812c commit d74c622

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

@@ -1061,6 +1063,8 @@ pub(crate) unsafe fn enzyme_rust_forward_diff(
10611063

10621064
EnzymeCreateForwardDiff(
10631065
logic_ref, // Logic
1066+
ptr::null(),
1067+
ptr::null(),
10641068
fnc,
10651069
ret_activity, // LLVM function, return type
10661070
input_activity.as_ptr(),
@@ -1123,6 +1127,8 @@ pub(crate) unsafe fn enzyme_rust_reverse_diff(
11231127

11241128
EnzymeCreatePrimalAndGradient(
11251129
logic_ref, // Logic
1130+
ptr::null(),
1131+
ptr::null(),
11261132
fnc,
11271133
ret_activity, // LLVM function, return type
11281134
input_activity.as_ptr(),
@@ -2818,6 +2824,8 @@ pub enum CDerivativeMode {
28182824
extern "C" {
28192825
fn EnzymeCreatePrimalAndGradient<'a>(
28202826
arg1: EnzymeLogicRef,
2827+
builderCtx: *const u8, // &'a Builder<'_>,
2828+
callerCtx: *const u8,// &'a Value,
28212829
todiff: &'a Value,
28222830
retType: CDIFFE_TYPE,
28232831
constant_args: *const CDIFFE_TYPE,
@@ -2841,6 +2849,8 @@ extern "C" {
28412849
extern "C" {
28422850
fn EnzymeCreateForwardDiff<'a>(
28432851
arg1: EnzymeLogicRef,
2852+
builderCtx: *const u8,// &'a Builder<'_>,
2853+
callerCtx: *const u8,// &'a Value,
28442854
todiff: &'a Value,
28452855
retType: CDIFFE_TYPE,
28462856
constant_args: *const CDIFFE_TYPE,

src/tools/enzyme

Submodule enzyme updated 82 files

0 commit comments

Comments
 (0)