Skip to content

Commit 1a2a4ed

Browse files
authored
Use new b2b kernels in evo2 jet tests (#985)
### Description Add `--use-b2b-causal-conv1d` arg to evo2 scripts in jet tests so they use accelerated kernels. Signed-off-by: Jared Wilber <jwilber@nvidia.com>
1 parent 7874355 commit 1a2a4ed

2 files changed

Lines changed: 2 additions & 0 deletions

File tree

ci/benchmarks/partial-conv/evo2_pretrain.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -79,6 +79,7 @@ script: |-
7979
--limit-val-batches=20 \
8080
--log-every-n-steps=50 \
8181
--val-check-interval=500 \
82+
--use-b2b-causal-conv1d \
8283
--create-tflops-callback \
8384
--create-tensorboard-logger \
8485
--result-dir=${tensorboard_dir} \

ci/benchmarks/perf/evo2_pretrain.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -76,6 +76,7 @@ script: |-
7676
--limit-val-batches=20 \
7777
--log-every-n-steps=50 \
7878
--val-check-interval=200 \
79+
--use-b2b-causal-conv1d \
7980
--create-tflops-callback \
8081
--create-tensorboard-logger \
8182
--result-dir=${tensorboard_dir} \

0 commit comments

Comments
 (0)