fix ci errors

Author: anupambhatnagar

beginner_source/hta_trace_diff_tutorial.rst

@@ -34,8 +34,8 @@ The `TraceDiff <https://hta.readthedocs.io/en/latest/source/api/trace_diff_api.h
 #. `visualize_duration_diff
    <https://hta.readthedocs.io/en/latest/source/api/trace_diff_api.html#hta.trace_diff.TraceDiff.visualize_duration_diff>`_
 
-The last two methods can be used to visualize various changes in counts and
-durations of CPU operators and GPU kernels using the output of the
+The last two methods can be used to visualize various changes in frequency and
+duration of CPU operators and GPU kernels, using the output of the
 ``compare_traces`` method.
 
 E.g. The top 10 operators with increase in frequency can be computed as
@@ -63,5 +63,5 @@ follows:
 
 For a detailed example of this feature see the `trace_diff_demo notebook
 <https://github.com/facebookresearch/HolisticTraceAnalysis/blob/main/examples/trace_diff_demo.ipynb>`_
-in the examples folder of the repo.
+in the examples folder of the repository.
 

en-wordlist.txt

@@ -554,3 +554,15 @@ pre
 otsu
 rgb
 histologically
+Conda
+CudaLaunchKernel
+CudaMemcpyAsync
+CudaMemsetAsync
+HTA
+bw
+conda
+cuda
+enqueuing
+jupyter
+memcpy
+memset

requirements.txt

@@ -20,7 +20,7 @@ bs4
 awscliv2==2.1.1
 flask
 spacy==3.4.1
-#ray[tune]==2.4.0
+ray[tune]==2.4.0
 tensorboard
 jinja2==3.0.3
 pytorch-lightning
@@ -32,9 +32,9 @@ nbformat>=4.2.0
 datasets
 transformers
 torchmultimodal-nightly # needs to be updated to stable as soon as it's avaialable
-#onnx
-#onnxscript
-#onnxruntime
+onnx
+onnxscript
+onnxruntime
 
 importlib-metadata==6.8.0
 
@@ -58,6 +58,6 @@ pyopengl
 gymnasium[mujoco]==0.27.0
 timm
 iopath
-#pygame==2.1.2
+pygame==2.1.2
 pycocotools
 semilearn==0.3.2