Function getSMVersion

plugin/common/bertCommon.h:94–101

{

inline int32_t getSMVersion()
{
    int32_t device{-1};
    PLUGIN_CHECK(cudaGetDevice(&device));
    cudaDeviceProp props;
    PLUGIN_CHECK(cudaGetDeviceProperties(&props, device));
    return nvinfer1::plugin::getTrtSMVersionDec(props.major, props.minor);
}

inline int32_t getMHAMaskPackedSize(int32_t smVersion, nvinfer1::DataType dataType, int32_t sequenceLength)
{