Skip to content

Commit

Permalink
Validate transformer building blocks tutorial builds against 2.6
Browse files Browse the repository at this point in the history
  • Loading branch information
mikaylagawarecki committed Dec 23, 2024
1 parent 39d3e6e commit 1706a1c
Show file tree
Hide file tree
Showing 2 changed files with 264 additions and 139 deletions.
12 changes: 6 additions & 6 deletions .jenkins/validate_tutorials_built.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@

NOT_RUN = [
"beginner_source/basics/intro", # no code
"beginner_source/introyt/introyt_index", # no code
"beginner_source/introyt/introyt_index", # no code
"beginner_source/onnx/intro_onnx",
"beginner_source/profiler",
"beginner_source/saving_loading_models",
Expand All @@ -25,9 +25,8 @@
"intermediate_source/mnist_train_nas", # used by ax_multiobjective_nas_tutorial.py
"intermediate_source/fx_conv_bn_fuser",
"intermediate_source/_torch_export_nightly_tutorial", # does not work on release
"intermediate_source/transformer_building_blocks", # does not work on release
"advanced_source/super_resolution_with_onnxruntime",
"advanced_source/usb_semisup_learn", # fails with CUDA OOM error, should try on a different worker
"advanced_source/usb_semisup_learn", # fails with CUDA OOM error, should try on a different worker
"prototype_source/fx_graph_mode_ptq_dynamic",
"prototype_source/vmap_recipe",
"prototype_source/torchscript_freezing",
Expand All @@ -50,11 +49,12 @@
"recipes_source/recipes/Captum_Recipe",
"intermediate_source/flask_rest_api_tutorial",
"intermediate_source/text_to_speech_with_torchaudio",
"intermediate_source/tensorboard_profiler_tutorial", # reenable after 2.0 release.
"intermediate_source/torch_export_tutorial", # reenable after 2940 is fixed.
"prototype_source/gpu_quantization_torchao_tutorial", # enable when 3194
"intermediate_source/tensorboard_profiler_tutorial", # reenable after 2.0 release.
"intermediate_source/torch_export_tutorial", # reenable after 2940 is fixed.
"prototype_source/gpu_quantization_torchao_tutorial", # enable when 3194
]


def tutorial_source_dirs() -> List[Path]:
return [
p.relative_to(REPO_ROOT).with_name(p.stem[:-7])
Expand Down
Loading

0 comments on commit 1706a1c

Please sign in to comment.