mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
73604eed0c
Follows #133067 Pull Request resolved: https://github.com/pytorch/pytorch/pull/133399 Approved by: https://github.com/Skylion007
19 lines
500 B
C++
19 lines
500 B
C++
#pragma once
|
|
|
|
#include <ATen/core/stack.h>
|
|
#include <torch/csrc/Export.h>
|
|
#include <torch/csrc/jit/codegen/fuser/fused_kernel.h>
|
|
#include <torch/csrc/jit/codegen/fuser/kernel_spec.h>
|
|
|
|
#include <cstdint>
|
|
|
|
namespace torch::jit::fuser {
|
|
|
|
// Runs the fusion associated with the key (see registerFusion() in interface.h)
|
|
// on the inputs taken from the given Stack.
|
|
TORCH_API bool runFusion(
|
|
const int64_t key,
|
|
Stack& stack,
|
|
std::string* code_out = nullptr);
|
|
|
|
} // namespace torch::jit::fuser
|