MLCompilerBridge
Tools for streamlining communication with ML models for compiler optimizations.
|
This file defines the PipeModelRunner class to interface with an external ML model during training and inference. More...
#include "MLModelRunner/PipeModelRunner.h"
#include "MLModelRunner/MLModelRunner.h"
#include "MLModelRunner/Utils/Debug.h"
#include <cstddef>
#include <cstring>
#include <fstream>
#include <iostream>
#include <string>
Go to the source code of this file.
Namespaces | |
namespace | MLBridge |
Macros | |
#define | DEBUG_TYPE "pipe-model-runner" |
This file defines the PipeModelRunner class to interface with an external ML model during training and inference.
The model is assumed to be running as an external process and the communication is done via 2 file descriptors using pipes.
Definition in file PipeModelRunner.cpp.
#define DEBUG_TYPE "pipe-model-runner" |
Definition at line 28 of file PipeModelRunner.cpp.