2
2
3
3
use libc:: { c_char, c_uint, size_t} ;
4
4
use rustc_ast:: expand:: autodiff_attrs:: DiffActivity ;
5
- use rustc_middle:: mir:: RuntimePhase ;
6
5
use tracing:: trace;
7
6
8
7
use super :: ffi:: * ;
@@ -78,6 +77,7 @@ pub(crate) unsafe fn enzyme_rust_forward_diff(
78
77
input_diffactivity : Vec < DiffActivity > ,
79
78
ret_diffactivity : DiffActivity ,
80
79
void_ret : bool ,
80
+ runtimeactivity : bool ,
81
81
) -> ( & Value , Vec < usize > ) {
82
82
let ret_activity = cdiffe_from ( ret_diffactivity) ;
83
83
assert ! ( ret_activity != CDIFFE_TYPE :: DFT_OUT_DIFF ) ;
@@ -127,8 +127,6 @@ pub(crate) unsafe fn enzyme_rust_forward_diff(
127
127
KnownValues : known_values. as_mut_ptr ( ) ,
128
128
} ;
129
129
130
- let runtimeactivity = false as u8 ;
131
-
132
130
trace ! ( "ret_activity: {}" , & ret_activity) ;
133
131
for i in & input_activity {
134
132
trace ! ( "input_activity i: {}" , & i) ;
@@ -146,7 +144,7 @@ pub(crate) unsafe fn enzyme_rust_forward_diff(
146
144
type_analysis, // type analysis struct
147
145
ret_primary_ret as u8 ,
148
146
CDerivativeMode :: DEM_ForwardMode , // return value, dret_used, top_level which was 1
149
- runtimeactivity,
147
+ runtimeactivity as u8 ,
150
148
1 , // free memory
151
149
1 , // vector mode width
152
150
Option :: None ,
@@ -166,6 +164,7 @@ pub(crate) unsafe fn enzyme_rust_reverse_diff(
166
164
fnc : & Value ,
167
165
rust_input_activity : Vec < DiffActivity > ,
168
166
ret_activity : DiffActivity ,
167
+ runtimeactivity : bool ,
169
168
) -> ( & Value , Vec < usize > ) {
170
169
let ( primary_ret, ret_activity) = match ret_activity {
171
170
DiffActivity :: Const => ( true , CDIFFE_TYPE :: DFT_CONSTANT ) ,
@@ -210,7 +209,6 @@ pub(crate) unsafe fn enzyme_rust_reverse_diff(
210
209
Return : ret_tt. inner ,
211
210
KnownValues : known_values. as_mut_ptr ( ) ,
212
211
} ;
213
- let runtimeactivity = false as u8 ;
214
212
215
213
trace ! ( "primary_ret: {}" , & primary_ret) ;
216
214
trace ! ( "ret_activity: {}" , & ret_activity) ;
@@ -231,7 +229,7 @@ pub(crate) unsafe fn enzyme_rust_reverse_diff(
231
229
primary_ret as u8 ,
232
230
diff_ret as u8 , //0
233
231
CDerivativeMode :: DEM_ReverseModeCombined , // return value, dret_used, top_level which was 1
234
- runtimeactivity,
232
+ runtimeactivity as u8 ,
235
233
1 , // vector mode width
236
234
1 , // free memory
237
235
Option :: None ,
@@ -667,11 +665,6 @@ pub mod Enzyme_AD {
667
665
static mut looseTypeAnalysis: c_void ;
668
666
static mut EnzymeInline : c_void ;
669
667
}
670
- pub fn set_runtime_activity_check ( check : bool ) {
671
- //unsafe {
672
- // EnzymeSetCLBool(std::ptr::addr_of_mut!(EnzymeRuntimeActivityCheck), check as u8);
673
- //}
674
- }
675
668
pub fn set_max_int_offset ( offset : u64 ) {
676
669
let offset = offset. try_into ( ) . unwrap ( ) ;
677
670
unsafe {
0 commit comments