Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Clear up neuron cache #3326

Merged
merged 2 commits into from
Sep 20, 2024
Merged

Clear up neuron cache #3326

merged 2 commits into from
Sep 20, 2024

Conversation

chen3933
Copy link
Contributor

@chen3933 chen3933 commented Sep 20, 2024

Description

Please read our CONTRIBUTING.md prior to creating your first pull request.

Please include a summary of the feature or issue being fixed. Please also include relevant motivation and context. List any dependencies that are required for this change.

Fixes #(issue)

Type of change

Please delete options that are not relevant.

  • Bug fix (non-breaking change which fixes an issue)
  • Breaking change (fix or feature that would cause existing functionality to not work as expected)
  • New feature (non-breaking change which adds functionality)
  • This change requires a documentation update

Feature/Issue validation/testing

Please describe the Unit or Integration tests that you ran to verify your changes and relevant result summary. Provide instructions so it can be reproduced.
Please also list any relevant details for your test configuration.

  • Run local test manually
python ./benchmarks/benchmark-ab.py --tmp_dir /tmp --report_location /tmp --config_properties ./benchmarks/config.properties --config /home/ubuntu/actions-runner/_work/serve/serve/model_json_config/neuronx/scripted_mode_opt_6.7b_neuronx_batch_1_w1_b1.json
...

Preparing local execution...
*Terminating any existing Torchserve instance ...
torchserve --stop
TorchServe has stopped.
*Setting up model store...
Directory /var/tmp/neuron-compile-cache/ exists. Clearing contents...
Cache cleared: /var/tmp/neuron-compile-cache/
*Starting local Torchserve instance...
Running: torchserve --start --model-store /tmp/model_store --enable-model-api --disable-token-auth --workflow-store /tmp/wf_store --ts-config /tmp/benchmark/conf/config.properties  > /tmp/benchmark/logs/model_metrics.log
torchserve --start --model-store /tmp/model_store --enable-model-api --disable-token-auth --workflow-store /tmp/wf_store --ts-config /tmp/benchmark/conf/config.properties  > /tmp/benchmark/logs/model_metrics.log
*Testing system health...
{
  "status": "Healthy"
}

Checklist:

  • Did you have fun?
  • Have you added tests that prove your fix is effective or that this feature works?
  • Has code been commented, particularly in hard-to-understand areas?
  • Have you made corresponding changes to the documentation?

@mreso mreso requested a review from namannandan September 20, 2024 19:22
@namannandan namannandan added this pull request to the merge queue Sep 20, 2024
Merged via the queue into pytorch:master with commit 7161c6f Sep 20, 2024
14 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants