src: gpu: intel: jit: pooling: avoid looping when init fails

hidefromkgb · hidefromkgb · commit a9d0007232b1 · 2024-05-24T18:44:22.000-07:00
diff --git a/src/gpu/intel/jit/pooling/gen_pooling.cpp b/src/gpu/intel/jit/pooling/gen_pooling.cpp
@@ -153,6 +153,7 @@ status_t gen_pooling_fwd_t::init(engine_t *engine) {
         try {
             kernel_ = make_kernel<pooling_kernel_t>(this, engine, cfg_,
                     "gen_pooling_fwd", kernel_info_, grf_mode_t::any, *pd());
+            break;
         } catch (const ngen::out_of_registers_exception &exc) {
             UNUSED(exc);
             ir_warning() << "loop too large: cut and retry!" << std::endl;
@@ -161,6 +162,10 @@ status_t gen_pooling_fwd_t::init(engine_t *engine) {
                 ir_error_not_expected() << "minimal loop too large!";
                 break;
             }
+        } catch (const std::exception &exc) {
+            ir_error_not_expected() << exc.what();
+            kernel_ = {};
+            break;
         }
     }
     set_version(cfg_.n_cuts());

Original file line number	Diff line number	Diff line change
`@@ -153,6 +153,7 @@ status_t gen_pooling_fwd_t::init(engine_t *engine) {`
`153`	`153`	`try {`
`154`	`154`	`kernel_ = make_kernel<pooling_kernel_t>(this, engine, cfg_,`
`155`	`155`	`"gen_pooling_fwd", kernel_info_, grf_mode_t::any, *pd());`
	`156`	`+ break;`
`156`	`157`	`} catch (const ngen::out_of_registers_exception &exc) {`
`157`	`158`	`UNUSED(exc);`
`158`	`159`	`ir_warning() << "loop too large: cut and retry!" << std::endl;`
`@@ -161,6 +162,10 @@ status_t gen_pooling_fwd_t::init(engine_t *engine) {`
`161`	`162`	`ir_error_not_expected() << "minimal loop too large!";`
`162`	`163`	`break;`
`163`	`164`	`}`
	`165`	`+ } catch (const std::exception &exc) {`
	`166`	`+ ir_error_not_expected() << exc.what();`
	`167`	`+ kernel_ = {};`
	`168`	`+ break;`
`164`	`169`	`}`
`165`	`170`	`}`
`166`	`171`	`set_version(cfg_.n_cuts());`