未验证 提交 4a06ecf6 编写于 作者: J Joe Mayer 提交者: GitHub

Updating autotune json default in docs. (#2476)

* Updating autotune default in docs.

* Running pre-commit.
上级 f74ee318
...@@ -7,12 +7,12 @@ ...@@ -7,12 +7,12 @@
#include "ATen/TensorUtils.h" #include "ATen/TensorUtils.h"
#include "ATen/cuda/CUDAContext.h" #include "ATen/cuda/CUDAContext.h"
#include "ATen/cuda/detail/IndexUtils.cuh" #include "ATen/cuda/detail/IndexUtils.cuh"
//#include "ATen/Type.h" // #include "ATen/Type.h"
#include "ATen/AccumulateType.h" #include "ATen/AccumulateType.h"
#include <iostream> #include <iostream>
//#include <helper_functions.h> // #include <helper_functions.h>
#if defined(__HIP_PLATFORM_HCC__) && HIP_VERSION > 305 #if defined(__HIP_PLATFORM_HCC__) && HIP_VERSION > 305
#include <hip/hip_cooperative_groups.h> #include <hip/hip_cooperative_groups.h>
#else #else
......
...@@ -652,8 +652,8 @@ Configuring the asynchronous I/O module for offloading parameter and optimizer s ...@@ -652,8 +652,8 @@ Configuring the asynchronous I/O module for offloading parameter and optimizer s
{ {
"autotuning": { "autotuning": {
"enabled": false, "enabled": false,
"results_dir": null, "results_dir": "autotuning_results",
"exps_dir": null, "exps_dir": "autotuning_exps",
"overwrite": false, "overwrite": false,
"metric": "throughput", "metric": "throughput",
"start_profile_step": 3, "start_profile_step": 3,
...@@ -678,15 +678,15 @@ Configuring the asynchronous I/O module for offloading parameter and optimizer s ...@@ -678,15 +678,15 @@ Configuring the asynchronous I/O module for offloading parameter and optimizer s
<i>**results_dir**</i>: [string] <i>**results_dir**</i>: [string]
| Description | Default | | Description | Default |
| -------------------------------------------------------------------------------------------------------------------------------- | ------- | | ------------------------------------------------------------------------------------------------------------------------------------- | --------------------- |
| Path to the autotuning experiment results directory. If None, "autotuning_results" under the training script launching path is used. | `null` | | Path to the autotuning experiment results directory. The default appears in the working directory from which Deepspeed was launched. | "autotuning_results" |
<i>**exps_dir**</i>: [string] <i>**exps_dir**</i>: [string]
| Description | Default | | Description | Default |
| ---------------------------------------------------------------------------------------------------------------------------------- | ------- | | ---------------------------------------------------------------------------------------------------------------------------------------- | ------------------ |
| Path to the auotuning experiment descriptions directory. If None, "autotuning_exps" under the train script launching path is used. | `null` | | Path to the auotuning experiment descriptions directory. The default appears in the working directory from which Deepspeed was launched. | "autotuning_exps" |
<i>**overwrite**</i>: [boolean] <i>**overwrite**</i>: [boolean]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册