Release - SuperBench v0.2.1 (#142)

__Description__
Cherry-pick bug fixes from v0.2.1 to main.

__Major Revisions__
* Fix bug of VGG models failed on A100 GPU with batch_size=128.
* Fix Ansible connection issue when running in localhost.
* Update version in packages and docs.

README.md

@@ -15,7 +15,7 @@
 
 __SuperBench__ is a validation and profiling tool for AI infrastructure.
 
-📢 [v0.2.0](https://github.com/microsoft/superbenchmark/releases/tag/v0.2.0) has been released!
+📢 [v0.2.1](https://github.com/microsoft/superbenchmark/releases/tag/v0.2.1) has been released!
 
 ## _Check [aka.ms/superbench](https://aka.ms/superbench) for more details._
 

docs/getting-started/installation.md

@@ -57,7 +57,7 @@ You can clone the source from GitHub and build it.
 :::note Note
 You should checkout corresponding tag to use release version, for example,
 
-`git clone -b v0.2.0 https://github.com/microsoft/superbenchmark`
+`git clone -b v0.2.1 https://github.com/microsoft/superbenchmark`
 :::
 
 ```bash

docs/getting-started/run-superbench.md

@@ -27,7 +27,7 @@ sb deploy -f remote.ini --host-password [password]
 :::note Note
 You should deploy corresponding Docker image to use release version, for example,
 
-`sb deploy -f local.ini -i superbench/superbench:v0.2.0-cuda11.1.1`
+`sb deploy -f local.ini -i superbench/superbench:v0.2.1-cuda11.1.1`
 :::
 
 ## Run

superbench/__init__.py

@@ -6,5 +6,5 @@
 Provide hardware and software benchmarks for AI systems.
 """
 
-__version__ = '0.2.0'
+__version__ = '0.2.1'
 __author__ = 'Microsoft'

superbench/benchmarks/model_benchmarks/pytorch_base.py

@@ -184,11 +184,13 @@ def _postprocess(self):
             )
             return False
 
-        del self._model
-        del self._optimizer
+        if self._gpu_available:
+            torch.cuda.synchronize()
         del self._target
-
-        torch.cuda.empty_cache()
+        del self._optimizer
+        del self._model
+        if self._gpu_available:
+            torch.cuda.empty_cache()
 
         return True
 

superbench/runner/playbooks/deploy.yaml

@@ -4,7 +4,7 @@
 
 - name: Context Preparation
   hosts: localhost
-  connection: localhost
+  connection: local
   gather_facts: false
   tasks:
     - name: Generating SSH Config

website/docusaurus.config.js

@@ -101,7 +101,7 @@ module.exports = {
     announcementBar: {
       id: 'supportus',
       content:
-        '📢 <a href="https://microsoft.github.io/superbenchmark/blog/release-sb-v0.2">v0.2</a> has been released! ' +
+        '📢 <a href="https://microsoft.github.io/superbenchmark/blog/release-sb-v0.2">v0.2.1</a> has been released! ' +
         '⭐️ If you like SuperBench, give it a star on <a target="_blank" rel="noopener noreferrer" href="https://github.com/microsoft/superbenchmark">GitHub</a>! ⭐️',
     },
     prism: {

website/package-lock.json

 {
   "name": "superbench-website",
-  "version": "0.2.0",
+  "version": "0.2.1",
   "lockfileVersion": 1,
   "requires": true,
   "dependencies": {

website/package.json

 {
   "name": "superbench-website",
-  "version": "0.2.0",
+  "version": "0.2.1",
   "private": true,
   "scripts": {
     "docusaurus": "docusaurus",