/** \brief Fusing linear patterns as single at::linear for easier pattern
|
* matching in later passes
|
*/
|
#pragma once
|
|
#include <torch/csrc/jit/ir.h>
|
|
namespace torch {
|
namespace jit {
|
|
/** \brief Match the at::linear pattern and fuse it into a single at::linear
|
* This pass fuse the addmm or matmul + add generated by JIT back to linear
|
* This pass can be deleted once the JIT can emit the aten::linear in the future
|
*/
|
TORCH_API void FuseLinear(std::shared_ptr<Graph>& graph);
|
} // namespace jit
|
} // namespace torch
|
