Skip to content

Commit 4c41ac8

Browse files
committed
Skip int8 model on GPU and merge from upstream
1 parent 44982d0 commit 4c41ac8

1 file changed

Lines changed: 8 additions & 0 deletions

File tree

benchmark/benchncnn.cpp

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -48,6 +48,14 @@ static ncnn::VkAllocator* g_staging_vkallocator = 0;
4848

4949
void benchmark(const char* comment, const std::vector<ncnn::Mat>& _in, const ncnn::Option& opt, bool fixed_path = true)
5050
{
51+
// Skip if int8 model name and using GPU
52+
if (opt.use_vulkan_compute && strstr(comment, "int8") != NULL)
53+
{
54+
if (!fixed_path)
55+
fprintf(stderr, "%20s skipped (int8+GPU not supported)\n", comment);
56+
return;
57+
}
58+
5159
g_blob_pool_allocator.clear();
5260
g_workspace_pool_allocator.clear();
5361

0 commit comments

Comments
 (0)