Merge branch 'main' into sycl_extension/zzq

Author: sekyondaMeta

.jenkins/validate_tutorials_built.py

@@ -50,6 +50,7 @@
     "intermediate_source/text_to_speech_with_torchaudio",
     "intermediate_source/tensorboard_profiler_tutorial", # reenable after 2.0 release.
     "advanced_source/semi_structured_sparse", # reenable after 3303 is fixed.
+    "intermediate_source/torchrec_intro_tutorial.py", #failing with 2.8 reenable after 3498
 ]
 
 def tutorial_source_dirs() -> List[Path]:

intermediate_source/per_sample_grads.py

@@ -169,7 +169,7 @@ def compute_loss(params, buffers, sample, target):
 # results of hand processing each one individually:
 
 for per_sample_grad, ft_per_sample_grad in zip(per_sample_grads, ft_per_sample_grads.values()):
-    assert torch.allclose(per_sample_grad, ft_per_sample_grad, atol=3e-3, rtol=1e-5)
+    assert torch.allclose(per_sample_grad, ft_per_sample_grad, atol=1.2e-1, rtol=1e-5)
 
 ######################################################################
 # A quick note: there are limitations around what types of functions can be