source/slang/slang-ir-pytorch-cpp-binding.h


1
2
3
4
5
6
7
8
9
10
11
12

#pragma once

namespace Slang
{
struct IRModule;
class DiagnosticSink;

void generatePyTorchCppBinding(IRModule* module, DiagnosticSink* sink);
void removeTorchKernels(IRModule* module);

}