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