Skip to content

Commit cb3efe3

Browse files
author
Github Executorch
committed
Replace ones with rand to workaround the crash from sdpa kernel
1 parent 6cb6af9 commit cb3efe3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

extension/benchmark/apple/Benchmark/Tests/GenericTests.mm

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ @implementation GenericTests
7878

7979
const auto sizes = tensor_meta->sizes();
8080
tensors.emplace_back(
81-
ones({sizes.begin(), sizes.end()}, tensor_meta->scalar_type()));
81+
rand({sizes.begin(), sizes.end()}, tensor_meta->scalar_type()));
8282
XCTAssertEqual(module->set_input(tensors.back(), index), Error::Ok);
8383
} break;
8484
default:

0 commit comments

Comments
 (0)