File tree Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Original file line number Diff line number Diff line change @@ -2,12 +2,14 @@ using DynamicPPL.TestUtils: DEMO_MODELS
22using  DynamicPPL. TestUtils. AD:  run_ad
33using  ADTypes:  AutoEnzyme
44using  Test:  @test , @testset 
5- import  Enzyme:  set_runtime_activity, Forward, Reverse
5+ import  Enzyme:  set_runtime_activity, Forward, Reverse, Const 
66using  ForwardDiff:  ForwardDiff  #  run_ad uses FD for correctness test
77
88ADTYPES =  Dict (
9-     " EnzymeForward" =>  AutoEnzyme (; mode= set_runtime_activity (Forward)),
10-     " EnzymeReverse" =>  AutoEnzyme (; mode= set_runtime_activity (Reverse)),
9+     " EnzymeForward" => 
10+         AutoEnzyme (; mode= set_runtime_activity (Forward), function_annotation= Const),
11+     " EnzymeReverse" => 
12+         AutoEnzyme (; mode= set_runtime_activity (Reverse), function_annotation= Const),
1113)
1214
1315@testset  " $ad_key " for  (ad_key, ad_type) in  ADTYPES
 
 
   
 
     
   
   
          
    
    
     
    
      
     
     
    You can’t perform that action at this time.
  
 
    
  
    
      
        
     
       
      
     
   
 
    
    
  
 
  
 
     
    
0 commit comments