MLCompilerBridge
Tools for streamlining communication with ML models for compiler optimizations.
|
PipeModelRunner class supporting communication via OS pipes between the compiler and an external ML agent. More...
#include "MLModelRunner/MLModelRunner.h"
#include "SerDes/TensorSpec.h"
#include "SerDes/baseSerDes.h"
#include "llvm/Support/FileSystem.h"
#include <fstream>
#include <memory>
#include <system_error>
Go to the source code of this file.
Classes | |
class | MLBridge::PipeModelRunner |
A MLModelRunner that asks for advice from an external agent, or host. More... | |
Namespaces | |
namespace | MLBridge |
PipeModelRunner class supporting communication via OS pipes between the compiler and an external ML agent.
Usage:
This supports both training and inference. Supports interleaved communication.
Definition in file PipeModelRunner.h.