MCPcopy Create free account
hub / github.com/NVIDIA/TensorRT / stgStream

Function stgStream

plugin/instanceNormalizationPlugin/instanceNormCommon.h:231–235  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

229
230template <typename T>
231DEVICE_FUNCTION void stgStream(T* gmem, int32_t (&src)[1])
232{
233 uint32_t tmp = src[0];
234 asm volatile("st.global.cs.s32 [%0], %1;" ::"l"((uint32_t*) gmem), "r"(tmp));
235}
236
237template <typename T>
238DEVICE_FUNCTION void stg(T* gmem, int32_t (&src)[2])

Callers

nothing calls this directly

Calls 2

fromFloatFunction · 0.85
stgFunction · 0.85

Tested by

no test coverage detected