From d64ee86a3130f8eeb75d09193c38c621d7565eba Mon Sep 17 00:00:00 2001 From: Yong He Date: Sun, 26 Mar 2023 13:59:11 -0700 Subject: Add PyTorch C++ binding generation. (#2734) * Add PyTorch C++ binding generation. * fix --------- Co-authored-by: Yong He --- source/slang/slang-syntax.cpp | 7 +++++++ 1 file changed, 7 insertions(+) (limited to 'source/slang/slang-syntax.cpp') diff --git a/source/slang/slang-syntax.cpp b/source/slang/slang-syntax.cpp index 27aba435f..1c2726551 100644 --- a/source/slang/slang-syntax.cpp +++ b/source/slang/slang-syntax.cpp @@ -530,6 +530,13 @@ Index getFilterCountImpl(const ReflectClassInfo& clsInfo, MemberFilterStyle filt matType->declRef = declRef; return matType; } + else if (magicMod->magicName == "TensorViewType") + { + SLANG_ASSERT(subst && subst->getArgs().getCount() == 1); + auto vecType = astBuilder->getOrCreate(ExtractGenericArgType(subst->getArgs()[0])); + vecType->declRef = declRef; + return vecType; + } else if (magicMod->magicName == "Texture") { SLANG_ASSERT(subst && subst->getArgs().getCount() >= 1); -- cgit v1.2.3