Skip to content

advanced.md

Latest commit

 

History

History
36 lines (27 loc) · 1.33 KB

advanced.md

File metadata and controls

36 lines (27 loc) · 1.33 KB

More advanced facts:

  • Workload create has two mutually exclusive ways to override the environment of a workload:

    • a --env flag to specify each environment variable separately. The format is:

      --env VARIABLE1=value --env VARIABLE2=value

    • a --env-file flag to allow specifying the container's environment from a file. Usage is the same as Docker's --env-file flag

    Example Env File:

    LIBTPU_INIT_ARGS=--my-flag=true --performance=high
MY_ENV_VAR=hello

  • Workload create accepts a --debug-dump-gcs flag which is a path to GCS bucket. Passing this flag sets the XLA_FLAGS='--xla_dump_to=/tmp/xla_dump/' and uploads hlo dumps to the specified GCS bucket for each worker.