diff options
| author | jsmall-nvidia <jsmall@nvidia.com> | 2020-03-25 16:45:56 -0400 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2020-03-25 16:45:56 -0400 |
| commit | b3e6f1b2cffa8def593e97a00576eeba0f947ebc (patch) | |
| tree | f953b64922bb3fe69ef1ac26bef0eda2741626d3 /source/slang/slang-emit-cuda.cpp | |
| parent | 28a0ca96a1ad2a3f0e09cc97b866f3b6338a09fa (diff) | |
Unroll target improvements (#1291)
* Add unroll support for CUDA, and preliminary for C++.
Document [unroll] support.
* Fix loop-unroll to run on CPU, and test on CPU and elsewhere.
Fix bug in emitting loop unroll condition.
* Improved comment.
* Added support for vk/glsl loop unrolling.
Diffstat (limited to 'source/slang/slang-emit-cuda.cpp')
| -rw-r--r-- | source/slang/slang-emit-cuda.cpp | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/source/slang/slang-emit-cuda.cpp b/source/slang/slang-emit-cuda.cpp index 91439d5d3..64cb240fc 100644 --- a/source/slang/slang-emit-cuda.cpp +++ b/source/slang/slang-emit-cuda.cpp @@ -341,6 +341,14 @@ void CUDASourceEmitter::emitCall(const HLSLIntrinsic* specOp, IRInst* inst, cons return Super::emitCall(specOp, inst, operands, numOperands, inOuterPrec); } +void CUDASourceEmitter::emitLoopControlDecorationImpl(IRLoopControlDecoration* decl) +{ + if (decl->getMode() == kIRLoopControl_Unroll) + { + m_writer->emit("#pragma unroll\n"); + } +} + bool CUDASourceEmitter::tryEmitInstExprImpl(IRInst* inst, const EmitOpInfo& inOuterPrec) { switch(inst->op) |