neural_infer.h

Go to the documentation of this file.

 
#ifndef OAKFIELD_NEURAL_INFER_H
#define OAKFIELD_NEURAL_INFER_H
 
#include "oakfield/operator_split.h"
 
#include <stdbool.h>
#include <stddef.h>
 
#ifdef __cplusplus
extern "C" {
#endif
 
struct SimContext;
 
#define SIM_NEURAL_MODEL_ID_MAX 63U
 
typedef struct SimNeuralInferenceRequest {
    const char *model_id;                          
    SimNeuralDeterminismPolicy determinism_policy; 
    SimNeuralDeviceRequirement device_requirement; 
    SimNeuralPrecisionMode precision_mode;         
    SimNeuralShapeConstraints shape_constraints;   
    bool normalize_input; 
    double input_scale;   
    double input_bias;    
    double output_scale;  
    double output_bias;   
    size_t step_index;    
    double dt;            
    double sim_time;      
} SimNeuralInferenceRequest;
 
typedef SimResult (*SimNeuralInferenceFn)(void *userdata, const SimField *input, SimField *output,
                                          const SimNeuralInferenceRequest *request);
 
typedef struct SimNeuralBaseConfig {
    size_t input_field;                          
    size_t output_field;                         
    char model_id[SIM_NEURAL_MODEL_ID_MAX + 1U]; 
    bool accumulate;                             
    bool scale_by_dt;                            
    bool normalize_input; 
    double input_scale;   
    double input_bias;    
    double output_scale;  
    double output_bias;   
    SimNeuralDeterminismPolicy determinism_policy; 
    SimNeuralDeviceRequirement device_requirement; 
    SimNeuralPrecisionMode precision_mode;         
    SimNeuralShapeConstraints shape_constraints;   
    SimNeuralInferenceFn inference_fn;             
    void *inference_userdata;                      
} SimNeuralBaseConfig;
 
typedef SimNeuralBaseConfig SimNeuralInferOperatorConfig;
 
SimResult sim_add_neural_infer_operator(struct SimContext *context,
                                        const SimNeuralInferOperatorConfig *config,
                                        size_t *out_index);
 
SimResult sim_neural_infer_config(struct SimContext *context, size_t operator_index,
                                  SimNeuralInferOperatorConfig *out_config);
 
SimResult sim_neural_infer_update(struct SimContext *context, size_t operator_index,
                                  const SimNeuralInferOperatorConfig *config);
 
#ifdef __cplusplus
}
#endif
 
#endif /* OAKFIELD_NEURAL_INFER_H */