diff options
| author | Yong He <yonghe@outlook.com> | 2023-04-13 09:49:22 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-04-13 09:49:22 -0700 |
| commit | 813892cd023e216f6f6560eb47566522d3a82609 (patch) | |
| tree | 07cbf8851e0c178cbc895be73e17e6340cc22685 /source | |
| parent | 352a460fc866998da5f45a8c117d891c51ab5a47 (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.cpp | 4 |
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)"); |
