@@ -193,9 +193,8 @@ inline void dtype_specialized_elementwise_fn_impl(
193
193
});
194
194
}
195
195
196
- template <typename CTYPE_COMPUTE, typename Op, typename ... Args>
196
+ template <typename CTYPE_COMPUTE, typename ... Args>
197
197
inline bool validate_elementwise_fn_inputs (
198
- const Op& compute_fun,
199
198
KernelRuntimeContext& ctx,
200
199
const Tensor& out,
201
200
SupportedTensorDtypes out_dtypes,
@@ -288,7 +287,7 @@ inline void apply_elementwise_fn_runtime_out_dtypes(
288
287
SupportedTensorDtypes out_dtypes,
289
288
Args... inputs) {
290
289
const bool inputs_valid = validate_elementwise_fn_inputs<CTYPE_COMPUTE>(
291
- compute_fun, ctx, out, out_dtypes, inputs...);
290
+ ctx, out, out_dtypes, inputs...);
292
291
if (!inputs_valid) {
293
292
return ;
294
293
}
@@ -313,7 +312,7 @@ inline void apply_elementwise_fn(
313
312
const Tensor& out,
314
313
Args... inputs) {
315
314
const bool inputs_valid = validate_elementwise_fn_inputs<CTYPE_COMPUTE>(
316
- compute_fun, ctx, out, out_dtypes, inputs...);
315
+ ctx, out, out_dtypes, inputs...);
317
316
if (!inputs_valid) {
318
317
return ;
319
318
}
0 commit comments