sglang_v0.5.2/pytorch_2.8.0/torch/nativert/graph/GraphPasses.h

12 lines
248 B
C++

#pragma once
#include <torch/nativert/graph/Graph.h>
namespace torch::nativert {
void selectScalarOverload(torch::nativert::Graph* graph);
std::string selectScalarOverloadName(const torch::nativert::Node& node);
} // namespace torch::nativert