Skip to content

Commit e2779ee

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

File tree

4 files changed

+4
-4
lines changed

4 files changed

+4
-4
lines changed

.github/workflows/android-perf.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -86,7 +86,7 @@ jobs:
8686
# Separate default values from the workflow dispatch. To ensure defaults are accessible
8787
# during scheduled runs and to provide flexibility for different defaults between
8888
# on-demand and periodic benchmarking.
89-
CRON_DEFAULT_MODELS: ${{ github.event_name == 'schedule' && 'stories110M,dl3,mv3,mv2,ic4,ic3,vit,meta-llama/Llama-3.2-1B' || 'stories110M' }}
89+
CRON_DEFAULT_MODELS: ${{ github.event_name == 'schedule' && 'stories110M,dl3,mv3,mv2,ic4,ic3,vit' || 'stories110M' }}
9090
CRON_DEFAULT_DEVICES: samsung_galaxy_s22
9191
CRON_DEFAULT_DELEGATES: ${{ github.event_name == 'schedule' && 'xnnpack,qnn' || 'xnnpack' }}
9292
run: |

.github/workflows/apple-perf.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,7 @@ jobs:
7676
# Separate default values from the workflow dispatch. To ensure defaults are accessible
7777
# during scheduled runs and to provide flexibility for different defaults between
7878
# on-demand and periodic benchmarking.
79-
CRON_DEFAULT_MODELS: ${{ github.event_name == 'schedule' && 'stories110M,mv3,mv2,ic4,ic3,resnet50,edsr,mobilebert,w2l,meta-llama/Llama-3.2-1B' || 'stories110M' }}
79+
CRON_DEFAULT_MODELS: ${{ github.event_name == 'schedule' && 'stories110M,mv3,mv2,ic4,ic3,resnet50,edsr,mobilebert,w2l' || 'stories110M' }}
8080
CRON_DEFAULT_DEVICES: apple_iphone_15
8181
CRON_DEFAULT_DELEGATES: ${{ github.event_name == 'schedule' && 'xnnpack,coreml,mps' || 'xnnpack' }}
8282
run: |

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:

extension/benchmark/apple/Benchmark/Tests/LLaMA/LLaMATests.mm

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,7 @@ @implementation LLaMATests
8787
tokensPerSecondMetric.tokenCount = 0;
8888
const auto status = runner->generate(
8989
"Once upon a time",
90-
128,
90+
50,
9191
[=](const std::string &token) {
9292
tokensPerSecondMetric.tokenCount++;
9393
},

0 commit comments

Comments
 (0)