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

larryliu0820 · larryliu0820 · commit daf808ec2404 · 2025-06-23T15:25:30.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/extension/tensor/test/tensor_ptr_maker_test.cpp b/extension/tensor/test/tensor_ptr_maker_test.cpp
@@ -565,8 +565,7 @@ TEST_F(TensorPtrMakerTest, CreateArangeTensorWithDoubleType) {
 TEST_F(TensorPtrMakerTest, CreateArangeTensorWithEmptyRange) {
   // End < start with positive step should error out
   EXPECT_DEATH(
-    arange(5, 0, 1),
-    "inferred dimension size must be positive, got 0");
+      arange(5, 0, 1), "inferred dimension size must be positive, got 0");
 }
 
 TEST_F(TensorPtrMakerTest, CreateArangeTensorWithValidSizes) {

Original file line number	Diff line number	Diff line change
`@@ -565,8 +565,7 @@ TEST_F(TensorPtrMakerTest, CreateArangeTensorWithDoubleType) {`
`565`	`565`	`TEST_F(TensorPtrMakerTest, CreateArangeTensorWithEmptyRange) {`
`566`	`566`	`// End < start with positive step should error out`
`567`	`567`	`EXPECT_DEATH(`
`568`		`- arange(5, 0, 1),`
`569`		`- "inferred dimension size must be positive, got 0");`
	`568`	`+ arange(5, 0, 1), "inferred dimension size must be positive, got 0");`
`570`	`569`	`}`
`571`	`570`
`572`	`571`	`TEST_F(TensorPtrMakerTest, CreateArangeTensorWithValidSizes) {`