MCPcopy Create free account
hub / github.com/ARM-software/ComputeLibrary / get_gemm_lws

Function get_gemm_lws

src/core/CL/DefaultLWSHeuristics.cpp:30–49  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

28namespace
29{
30cl::NDRange get_gemm_lws(size_t gws_x, size_t gws_y, size_t gws_z)
31{
32 ARM_COMPUTE_UNUSED(gws_y);
33
34 if (gws_z != 1)
35 {
36 return cl::NDRange(4, 4, 2);
37 }
38 else
39 {
40 if (gws_x > 256)
41 {
42 return cl::NDRange(2, 16, 1);
43 }
44 else
45 {
46 return cl::NDRange(32, 4, 1);
47 }
48 }
49}
50
51cl::NDRange get_winograd_lws(size_t gws_x, size_t gws_y, size_t gws_z)
52{

Callers 1

get_default_lws_for_typeFunction · 0.85

Calls 1

NDRangeClass · 0.50

Tested by

no test coverage detected