From 813892cd023e216f6f6560eb47566522d3a82609 Mon Sep 17 00:00:00 2001 From: Yong He Date: Thu, 13 Apr 2023 09:49:22 -0700 Subject: Set sharedMem argument to 0 when launching cuda kernel. (#2799) Co-authored-by: Yong He --- source/slang/slang-emit-torch.cpp | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) (limited to 'source') diff --git a/source/slang/slang-emit-torch.cpp b/source/slang/slang-emit-torch.cpp index 819a6a136..bdb650607 100644 --- a/source/slang/slang-emit-torch.cpp +++ b/source/slang/slang-emit-torch.cpp @@ -94,9 +94,7 @@ void TorchCppSourceEmitter::emitInstStmtImpl(IRInst* inst) m_writer->emit(", "); // shared mem - m_writer->emit("slangGetCudaKernelSharedMemSize((const void*)("); - emitOperand(inst->getOperand(0), getInfo(EmitOp::General)); - m_writer->emit(")), "); + m_writer->emit("0, "); // stream m_writer->emit("((cudaStream_t)"); -- cgit v1.2.3