[DEVICE] Use noinline for LLGenericOp only on gfx950 (#1849)
This commit is contained in:
zatwierdzone przez
GitHub
rodzic
44121db890
commit
c3b8de4ec8
@@ -432,7 +432,11 @@ private:
|
||||
}
|
||||
|
||||
template <int RECV, int SEND, int SrcBuf, int DstBuf>
|
||||
#if defined(__gfx950__)
|
||||
__device__ __attribute__((noinline)) void LLGenericOp(intptr_t srcIx, intptr_t dstIx, int nelem, bool postOp) {
|
||||
#else
|
||||
__device__ void LLGenericOp(intptr_t srcIx, intptr_t dstIx, int nelem, bool postOp) {
|
||||
#endif
|
||||
constexpr int SRC = SrcBuf != -1 ? 1 : 0;
|
||||
constexpr int DST = DstBuf != -1 ? 1 : 0;
|
||||
T *srcElts = SrcBuf == -1 ? nullptr : userBufs[SrcBuf] + srcIx;
|
||||
|
||||
Reference in New Issue
Block a user