summaryrefslogtreecommitdiffstats
path: root/source
diff options
context:
space:
mode:
authorYong He <yonghe@outlook.com>2023-04-13 09:49:22 -0700
committerGitHub <noreply@github.com>2023-04-13 09:49:22 -0700
commit813892cd023e216f6f6560eb47566522d3a82609 (patch)
tree07cbf8851e0c178cbc895be73e17e6340cc22685 /source
parent352a460fc866998da5f45a8c117d891c51ab5a47 (diff)
Set sharedMem argument to 0 when launching cuda kernel. (#2799)
Co-authored-by: Yong He <yhe@nvidia.com>
Diffstat (limited to 'source')
-rw-r--r--source/slang/slang-emit-torch.cpp4
1 files changed, 1 insertions, 3 deletions
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)");