• Wu Zhiwen's avatar
    dnn/Vulkan: skip heavy convolution task · 33c9d57c
    Wu Zhiwen authored
    This is a workaround for GPU hang on heavy convolution workload (> 10 GFLOPS).
    e.g. ResNet101_DUC_HDC
    
    For the long time task, vkWaitForFences() return without error but next call on
    vkQueueSubmit() return -4, i.e. "VK_ERROR_DEVICE_LOST" and driver reports GPU hang.
    
    Need more investigation on root cause of GPU hang and need to optimize convolution shader
    to reduce process time.
    33c9d57c
dnn.cpp 129 KB