For div operator and for generic case, added get_compute type function to support integer inputs and float output

cmadhira@cadence.com · cmadhira@cadence.com · commit cdca6ca114f0 · 2025-01-03T11:52:54.000+05:30
diff --git a/backends/cadence/fusion_g3/operators/op_div.cpp b/backends/cadence/fusion_g3/operators/op_div.cpp
@@ -220,6 +220,8 @@ Tensor& div_out(KernelRuntimeContext& ctx,
     }
     else
     {
+        ScalarType common_type = get_common_type(a.scalar_type(), b.scalar_type());
+        ScalarType compute_type = torch::executor::native::utils::get_compute_type(common_type);
 
         ET_SWITCH_FLOAT_TYPES(compute_type, ctx, op_name, CTYPE_COMPUTE, [&]() {
             torch::executor::native::utils::
@@ -540,6 +542,8 @@ Tensor& div_scalar_out(KernelRuntimeContext& ctx,
     }
     else
     {
+        ScalarType common_type = executorch::runtime::isFloatingType(a.scalar_type()) ? a.scalar_type() : ScalarType::Float;
+        ScalarType compute_type = torch::executor::native::utils::get_compute_type(common_type);
         ET_SWITCH_FLOAT_TYPES(compute_type, ctx, op_name, CTYPE_COMPUTE, [&]() {
             const CTYPE_COMPUTE val_b = torch::executor::native::utils::
             scalar_to<CTYPE_COMPUTE>(b);

Original file line number	Diff line number	Diff line change
`@@ -220,6 +220,8 @@ Tensor& div_out(KernelRuntimeContext& ctx,`
`220`	`220`	`}`
`221`	`221`	`else`
`222`	`222`	`{`
	`223`	`+ ScalarType common_type = get_common_type(a.scalar_type(), b.scalar_type());`
	`224`	`+ ScalarType compute_type = torch::executor::native::utils::get_compute_type(common_type);`
`223`	`225`
`224`	`226`	`ET_SWITCH_FLOAT_TYPES(compute_type, ctx, op_name, CTYPE_COMPUTE, [&]() {`
`225`	`227`	`torch::executor::native::utils::`
`@@ -540,6 +542,8 @@ Tensor& div_scalar_out(KernelRuntimeContext& ctx,`
`540`	`542`	`}`
`541`	`543`	`else`
`542`	`544`	`{`
	`545`	`+ ScalarType common_type = executorch::runtime::isFloatingType(a.scalar_type()) ? a.scalar_type() : ScalarType::Float;`
	`546`	`+ ScalarType compute_type = torch::executor::native::utils::get_compute_type(common_type);`
`543`	`547`	`ET_SWITCH_FLOAT_TYPES(compute_type, ctx, op_name, CTYPE_COMPUTE, [&]() {`
`544`	`548`	`const CTYPE_COMPUTE val_b = torch::executor::native::utils::`
`545`	`549`	`scalar_to<CTYPE_COMPUTE>(b);`