Update on "[llm] Add arange() tensor maker API"

larryliu0820 · larryliu0820 · commit 0e14656e84a5 · 2025-06-24T21:57:48.000-07:00
As titled. `arange()` taking a `sizes` argument to be able to take custom tensor sizes. Differential Revision: [D77184741](https://our.internmc.facebook.com/intern/diff/D77184741/) [ghstack-poisoned]
diff --git a/kernels/portable/cpu/util/arange_util.cpp b/kernels/portable/cpu/util/arange_util.cpp
@@ -38,11 +38,13 @@ void arange_out_impl(
     double end,
     double step,
     Tensor& out) {
+  (void)ctx;
   Tensor::SizesType numel = compute_arange_out_size(start, end, step);
   ET_ARANGE_IMPL(ctx, start, numel, step, out, "arange.start_out");
 }
 
 void arange_out_impl(KernelRuntimeContext& ctx, double end, Tensor& out) {
+  (void)ctx;
   ET_ARANGE_IMPL(ctx, 0.0, end, 1.0, out, "arange.out");
 }
 

Original file line number	Diff line number	Diff line change
`@@ -38,11 +38,13 @@ void arange_out_impl(`
`38`	`38`	`double end,`
`39`	`39`	`double step,`
`40`	`40`	`Tensor& out) {`
	`41`	`+ (void)ctx;`
`41`	`42`	`Tensor::SizesType numel = compute_arange_out_size(start, end, step);`
`42`	`43`	`ET_ARANGE_IMPL(ctx, start, numel, step, out, "arange.start_out");`
`43`	`44`	`}`
`44`	`45`
`45`	`46`	`void arange_out_impl(KernelRuntimeContext& ctx, double end, Tensor& out) {`
	`47`	`+ (void)ctx;`
`46`	`48`	`ET_ARANGE_IMPL(ctx, 0.0, end, 1.0, out, "arange.out");`
`47`	`49`	`}`
`48`	`50`