diff options
| author | Pablo Marquez Tello <pablo.tello@arm.com> | 2023-12-05 15:44:50 +0000 |
|---|---|---|
| committer | Pablo Marquez Tello <pablo.tello@arm.com> | 2023-12-06 15:05:15 +0000 |
| commit | 17e116e90e6b962a09c133c646b6ad7884e94693 (patch) | |
| tree | 7c0bba172a510fb0793d2052b3ce5f1d3174abc9 /src/cpu/operators/CpuWinogradConv2d.cpp | |
| parent | fadc9b1e0bba90d6a91beb65466b2a0895b3a5e4 (diff) | |
| download | ComputeLibrary-17e116e90e6b962a09c133c646b6ad7884e94693.tar.gz | |
Revert "thread_local _custom_scheduler"
This reverts commit ded5b182675e3166e947a8eb637b5b1e925816ab.
Resolves COMPMID-6735
Signed-off-by: Pablo Marquez Tello <pablo.tello@arm.com>
Change-Id: I9b69ca1ec80a671171d3f52081c4b8c61a676617
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/10838
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: <felixjohnny.thomasmathibalan@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Benchmark: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/cpu/operators/CpuWinogradConv2d.cpp')
| -rw-r--r-- | src/cpu/operators/CpuWinogradConv2d.cpp | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/cpu/operators/CpuWinogradConv2d.cpp b/src/cpu/operators/CpuWinogradConv2d.cpp index 1fb6d33a61..e4bcdc0b64 100644 --- a/src/cpu/operators/CpuWinogradConv2d.cpp +++ b/src/cpu/operators/CpuWinogradConv2d.cpp @@ -103,7 +103,7 @@ bool get_winograd_kernel_implementation(const ITensorInfo Tensor4DShape in_shape{internal_get_shape(src)}; Tensor4DShape out_shape{internal_get_shape(dst)}; Tensor4DShape kernel_shape{internal_get_shape(weights)}; - uint32_t nthreads = NEScheduler::num_threads(); + uint32_t nthreads = NEScheduler::get().num_threads(); // Get configuration arguments for Winograd winograd_cfg.output_rows = 0; winograd_cfg.output_cols = 0; @@ -183,7 +183,7 @@ void CpuWinogradConv2d::configure(const ITensorInfo *src, ARM_COMPUTE_LOG_PARAMS(src, weights, biases, dst, conv_info, act_info, enable_fast_math); ARM_COMPUTE_UNUSED(biases); const DataType data_type = src->data_type(); - uint32_t nthreads = NEScheduler::num_threads(); + uint32_t nthreads = NEScheduler::get().num_threads(); _data_layout = src->data_layout(); const Tensor4DShape kernel_shape{internal_get_shape(weights)}; @@ -361,7 +361,7 @@ void CpuWinogradConv2d::run(ITensorPack &tensors) auto output = tensors.get_tensor(ACL_DST); Window win; - const uint32_t nthreads = NEScheduler::num_threads(); + const uint32_t nthreads = NEScheduler::get().num_threads(); // The Winograd transform implementation does fine-grain threading inside the transforms. Just pass thread_id and nthreads. win.set(Window::DimX, Window::Dimension(0, nthreads, 1)); |