{
  "backup_data_uid": "c01331d6b4234a23",
  "build_compiler_vars": {},
  "compile_deps": {
    "library": {
      "local": "yes",
      "name": "NNTest library",
      "sort": 2,
      "tags": "lib,nntest"
    }
  },
  "data_name": "lstm-pre",
  "main_language": "python",
  "no_compile": "yes",
  "no_target_file": "yes",
  "only_for_target_os_tags": [
    "linux"
  ],
  "pass_env_to_resolve": "yes",
  "print_files_after_run": [
    "tmp-stderr.tmp"
  ],
  "process_in_tmp": "yes",
  "program": "yes",
  "run_cmds": {
    "default": {
      "add_dataset_file_to_env": "yes",
      "dataset_tags": [
        "dataset",
        "nntest",
        "tensor-lstm",
        "pre"
      ],
      "ignore_return_code": "no",
      "run_time": {
        "fine_grain_timer_file": "tmp-ck-timer.json",
        "output_invariant_of_repeat": "yes",
        "post_process_cmds": [
          "python $#ck_take_from_{script:4ae4f08be053a2b8}#$process.py"
        ],
        "post_process_via_ck": "yes",
        "pre_process_via_ck": {
          "data_uoa": "4ae4f08be053a2b8",
          "module_uoa": "script",
          "script_name": "process"
        },
        "run_cmd_main": "$<<CK_ENV_COMPILER_PYTHON_FILE>>$ ../lstm.py",
        "run_cmd_out1": "tmp-stdout.tmp",
        "run_cmd_out2": "tmp-stderr.tmp",
        "run_correctness_extra_keys_from_dataset_file_json": "yes",
        "run_correctness_output_files": [
          "tmp-ck-output.json"
        ],
        "run_correctness_vars": [
          "CK_IN_SHAPE_N",
          "CK_SEED"
        ],
        "run_input_files": [],
        "run_output_files": [
          "tmp-ck-timer.json",
          "tmp-ck-output.bin",
          "tmp-ck-output.json"
        ]
      }
    }
  },
  "run_deps": {
    "library": {
      "local": "yes",
      "name": "PyTorch",
      "sort": 1,
      "tags": "lib,pytorch"
    },
    "python": {
      "local": "yes",
      "name": "Python interpreter",
      "sort": 0,
      "tags": "compiler,python"
    },
    "weights": {
      "enable_if_env": [
        {
          "CK_LSTM_PYTORCH_WEIGHTS": [
            1,
            "1",
            "YES",
            "yes"
          ]
        }
      ],
      "local": "yes",
      "name": "PyTorch weights",
      "sort": 10,
      "tags": "model,pytorch"
    },
    "plugins-pre-pytorch": {
      "local": "yes",
      "name": "Pytorch LSTM RNNT Pre plugin",
      "sort": 120,
      "tags": "nntest,plugin,rnnt-pre,pytorch",
      "enable_if_env": [
        { "CK_RNNT_PRE_BACKEND": ["pytorch"] }
      ],
      "update_tags_if_env" : {
        "orig": [ { "CK_RNNT_PRE": "orig" } ],
        "orig-quant": [ { "CK_RNNT_PRE": "orig-quant" } ],
        "naive": [ { "CK_RNNT_PRE": "naive" } ]
      }
    },
    "plugins-pre-onnx": {
      "local": "yes",
      "name": "ONNX LSTM RNNT Pre plugin",
      "sort": 120,
      "tags": "nntest,plugin,rnnt-pre,onnx",
      "enable_if_env": [
        { "CK_RNNT_PRE_BACKEND": ["onnx"] }
      ],
      "update_tags_if_env" : {
        "orig": [ { "CK_RNNT_PRE": "orig" } ],
        "naive": [ { "CK_RNNT_PRE": "naive" } ]
      }
    },
    "plugins-pre-glow": {
      "local": "yes",
      "name": "GLOW LSTM RNNT Pre plugin",
      "sort": 120,
      "tags": "nntest,plugin,rnnt-pre,glow",
      "enable_if_env": [
        { "CK_RNNT_PRE_BACKEND": ["glow"] }
      ],
      "update_tags_if_env" : {
        "orig": [ { "CK_RNNT_PRE": "orig" } ],
        "naive": [ { "CK_RNNT_PRE": "naive" } ]
      }
    }
  },
  "run_vars": {
    "CK_ABS_DIFF_THRESHOLD": 0.001,
    "CK_IN_SHAPE_N": 1,
    "CK_OUT_RAW_DATA": "tmp-ck-output.bin",
    "CK_RNNT_PRE_BACKEND": "pytorch",
    "CK_RNNT_PRE": "orig"
  },
  "source_files": [
    "lstm.py"
  ],
  "species": [
    "6bd431f979ccac7b"
  ],
  "tags": [
    "nntest",
    "lstm",
    "pytorch",
    "float32"
  ],
  "version": "1.0.0"
}