gpu: ocl: concat: disable batching with scales

oneapi-src · Dec 9, 2022 · 1eac450 · 1eac450
1 parent 8aab73f
commit 1eac450
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/gpu/ocl/multi_concat.hpp b/src/gpu/ocl/multi_concat.hpp
@@ -48,7 +48,9 @@ struct multi_concat_t : public gpu_primitive_t {
         }
 
         status_t init(engine_t *engine) {
-            if (max_batch_size() == 0) return status::unimplemented;
+            bool ok = max_batch_size() != 0 && attr()->has_default_values()
+                    && set_default_params() == status::success;
+            if (!ok) return status::unimplemented;
 
             auto n_batches = utils::div_up(n_inputs(), max_batch_size());
             concat_pds_.resize(n_batches);