summaryrefslogtreecommitdiff
path: root/source/slang/slang-ir-pytorch-cpp-binding.h
blob: c35b6a8ebff0f0dd440c75811503e188dd5e1ea8 (plain)
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);

}