├── .flake8 ├── .gitignore ├── .pydocstyle ├── .travis.yml ├── ALPHA.md ├── CONTRIBUTING.rst ├── Jenkinsfile ├── LICENSE ├── README.rst ├── codecov.yml ├── conftest.py ├── debug ├── README.md ├── __main__.py ├── bupdb.py ├── butest.py ├── cfg.py ├── cmd.py ├── do_inject.py ├── gprint.py ├── graph.css ├── inject.py ├── myia.css ├── run.py ├── steps.py ├── test_template.html ├── tools.py ├── trace.py └── typ.py ├── devdoc.md ├── docs ├── Makefile ├── _static │ └── .gitkeep ├── conf.py ├── gradient.ipynb └── index.rst ├── environment.yml ├── examples ├── lstm.py ├── mlp.py ├── policy_gradient.py ├── rnn.py └── vae.py ├── myia ├── __init__.py ├── abstract │ ├── __init__.py │ ├── aliasing.py │ ├── amerge.py │ ├── data.py │ ├── infer.py │ ├── loop.py │ ├── macro.py │ ├── ref.py │ ├── to_abstract.py │ └── utils.py ├── api.py ├── classes.py ├── compile │ ├── __init__.py │ ├── backends │ │ ├── __init__.py │ │ ├── prim_groups.py │ │ └── python │ │ │ ├── __init__.py │ │ │ ├── implementations.py │ │ │ └── python.py │ ├── cconv.py │ ├── transform.py │ ├── utils.py │ └── vm.py ├── debug │ ├── finite_diff.py │ ├── label.py │ ├── traceback.py │ └── utils.py ├── frontends │ ├── __init__.py │ └── abstract_types.py ├── grad.py ├── graph_utils.py ├── hypermap.py ├── info.py ├── ir │ ├── __init__.py │ ├── anf.py │ ├── clone.py │ ├── manager.py │ ├── metagraph.py │ └── utils.py ├── lib.py ├── monomorphize.py ├── operations │ ├── __init__.py │ ├── gen.py │ ├── macro_abstract_array.py │ ├── macro_apply.py │ ├── macro_array_len.py │ ├── macro_call_object.py │ ├── macro_conv2d_grad_input.py │ ├── macro_dict_values.py │ ├── macro_dtype.py │ ├── macro_einsum.py │ ├── macro_embed.py │ ├── macro_getattr.py │ ├── macro_grad.py │ ├── macro_hasattr.py │ ├── macro_is.py │ ├── macro_isinstance.py │ ├── macro_make_list.py │ ├── macro_resolve.py │ ├── macro_to_scalar_type.py │ ├── macro_tuple_len.py │ ├── macro_typeof.py │ ├── macro_universal.py │ ├── macro_user_switch.py │ ├── op_array_getitem_wrap.py │ ├── op_array_reduce_dim.py │ ├── op_enumerate.py │ ├── op_full.py │ ├── op_gadd.py │ ├── op_hyper_map.py │ ├── op_is_not.py │ ├── op_range.py │ ├── op_slice.py │ ├── op_value_and_grad.py │ ├── op_zeros_like.py │ ├── op_zip.py │ ├── ops_array.py │ ├── ops_bool.py │ ├── ops_dunder.py │ ├── ops_nil.py │ ├── ops_scalar.py │ ├── ops_string.py │ ├── ops_tuple.py │ ├── ops_universe.py │ ├── prim_J.py │ ├── prim_Jinv.py │ ├── prim_argmax.py │ ├── prim_array_cast.py │ ├── prim_array_getitem.py │ ├── prim_array_map.py │ ├── prim_array_max.py │ ├── prim_array_reduce.py │ ├── prim_array_scan.py │ ├── prim_array_setitem.py │ ├── prim_array_to_scalar.py │ ├── prim_bool_and.py │ ├── prim_bool_eq.py │ ├── prim_bool_not.py │ ├── prim_bool_or.py │ ├── prim_broadcast_shape.py │ ├── prim_casttag.py │ ├── prim_concat.py │ ├── prim_conv2d.py │ ├── prim_conv2d_weight_grad.py │ ├── prim_conv_transpose2d.py │ ├── prim_dict_getitem.py │ ├── prim_dict_setitem.py │ ├── prim_distribute.py │ ├── prim_dot.py │ ├── prim_env_add.py │ ├── prim_env_getitem.py │ ├── prim_env_setitem.py │ ├── prim_extract_kwarg.py │ ├── prim_gather.py │ ├── prim_hastag.py │ ├── prim_hastype.py │ ├── prim_identity.py │ ├── prim_invert_permutation.py │ ├── prim_make_dict.py │ ├── prim_make_exception.py │ ├── prim_make_handle.py │ ├── prim_make_kwarg.py │ ├── prim_make_record.py │ ├── prim_make_tuple.py │ ├── prim_max_pool2d.py │ ├── prim_max_pool2d_grad.py │ ├── prim_partial.py │ ├── prim_raise_.py │ ├── prim_random_initialize.py │ ├── prim_random_uint32.py │ ├── prim_record_getitem.py │ ├── prim_record_setitem.py │ ├── prim_reshape.py │ ├── prim_return_.py │ ├── prim_scalar_abs.py │ ├── prim_scalar_add.py │ ├── prim_scalar_bit_and.py │ ├── prim_scalar_bit_lshift.py │ ├── prim_scalar_bit_not.py │ ├── prim_scalar_bit_or.py │ ├── prim_scalar_bit_rshift.py │ ├── prim_scalar_bit_xor.py │ ├── prim_scalar_cast.py │ ├── prim_scalar_cos.py │ ├── prim_scalar_div.py │ ├── prim_scalar_eq.py │ ├── prim_scalar_exp.py │ ├── prim_scalar_floor.py │ ├── prim_scalar_ge.py │ ├── prim_scalar_gt.py │ ├── prim_scalar_le.py │ ├── prim_scalar_log.py │ ├── prim_scalar_lt.py │ ├── prim_scalar_max.py │ ├── prim_scalar_mod.py │ ├── prim_scalar_mul.py │ ├── prim_scalar_ne.py │ ├── prim_scalar_pow.py │ ├── prim_scalar_sign.py │ ├── prim_scalar_sin.py │ ├── prim_scalar_sub.py │ ├── prim_scalar_tan.py │ ├── prim_scalar_tanh.py │ ├── prim_scalar_to_array.py │ ├── prim_scalar_trunc.py │ ├── prim_scalar_uadd.py │ ├── prim_scalar_usub.py │ ├── prim_scatter.py │ ├── prim_scatter_add.py │ ├── prim_shape.py │ ├── prim_split.py │ ├── prim_stop_gradient.py │ ├── prim_string_eq.py │ ├── prim_switch.py │ ├── prim_tagged.py │ ├── prim_take.py │ ├── prim_take_grad_inp.py │ ├── prim_transpose.py │ ├── prim_tuple_getitem.py │ ├── prim_tuple_setitem.py │ ├── prim_universe_getitem.py │ ├── prim_universe_setitem.py │ ├── prim_unsafe_static_cast.py │ ├── primitives.py │ └── utils.py ├── opt │ ├── __init__.py │ ├── cse.py │ ├── dde.py │ ├── lib.py │ ├── opt.py │ └── rewrite.py ├── parser.py ├── pipeline │ ├── __init__.py │ ├── pipeline.py │ ├── resources.py │ ├── standard.py │ └── steps.py ├── public_api.py ├── simplify_types.py ├── testing │ ├── __init__.py │ ├── common.py │ └── multitest.py ├── utils │ ├── __init__.py │ ├── env.py │ ├── errors.py │ ├── get_fields.py │ ├── intern.py │ ├── merge.py │ ├── misc.py │ ├── orderedset.py │ ├── partial.py │ ├── trace.py │ ├── unify.py │ ├── universe.py │ └── variables.py ├── validate.py ├── vm.py └── xtype.py ├── myia_backend_pytorch ├── README.rst ├── cpu-extras.conda ├── environment-cpu.yml ├── environment-gpu.yml ├── gpu-extras.conda ├── myia_backend_pytorch │ ├── __init__.py │ ├── pytorch.py │ └── pytorch_conv_grad.py ├── myia_tests_backend_pytorch │ └── __init__.py ├── poetry.lock ├── pyproject.toml ├── scripts │ └── gen_conda_env_file.sh └── tests │ └── test_pytorch_rng.py ├── myia_backend_relay ├── README.rst ├── environment.yml ├── myia_backend_relay │ ├── __init__.py │ ├── relay.py │ ├── relay_helpers.py │ └── relay_philox.py ├── myia_tests_backend_relay │ └── __init__.py ├── poetry.lock ├── pyproject.toml ├── relay-extras.conda ├── scripts │ └── gen_conda_env_file.sh └── tests │ ├── compile │ └── test_relay.py │ ├── test_compile.py │ └── test_relay_rng.py ├── myia_frontend_pytorch ├── README.rst ├── cpu-extras.conda ├── environment-cpu.yml ├── environment-gpu.yml ├── gpu-extras.conda ├── myia_frontend_pytorch │ ├── __init__.py │ ├── pytorch.py │ └── pytorch_abstract_types.py ├── poetry.lock ├── pyproject.toml ├── scripts │ └── gen_conda_env_file.sh └── tests │ └── test_validate.py ├── myia_utils ├── README.rst ├── myia_utils │ ├── __init__.py │ └── update_env.py ├── poetry.lock └── pyproject.toml ├── poetry.lock ├── pyproject.toml ├── pytest.ini ├── scripts ├── build-doc.sh ├── check.sh ├── ci-install.sh ├── composite.py.template ├── consolidate-coverage.sh ├── gen_conda_env_file.sh ├── infprim.py.template ├── macro.py.template ├── new_operation.py ├── op.py.template ├── prim.py.template ├── regen.py └── test_op.py.template └── tests ├── __init__.py ├── compile ├── __init__.py ├── test_backend.py └── test_cconv.py ├── composite ├── __init__.py ├── examples │ ├── __init__.py │ ├── operations.py │ ├── prim_composite_full.py │ ├── prim_composite_simple.py │ └── primitives.py └── test_composite_examples.py ├── debug └── test_debug_utils.py ├── frontends ├── __init__.py ├── test_all_ops.py ├── test_frontend.py └── test_pytorch.py ├── ir ├── __init__.py ├── test_anf.py ├── test_clone.py ├── test_ir_utils.py └── test_manager.py ├── operations ├── __init__.py ├── test_hyper_map.py ├── test_macro_dtype.py ├── test_macro_einsum.py ├── test_operations.py ├── test_ops_dunder.py ├── test_raise_assert.py └── test_universe.py ├── opt ├── __init__.py ├── test_lib.py ├── test_opt.py └── test_rewrite.py ├── prim ├── __init__.py └── test_py_implementations.py ├── python_backend ├── .pdbrc ├── test_frontend_compilation.py ├── test_graph_compilation.py ├── test_python_debug.py ├── test_python_pdb.py └── test_python_rng.py ├── test_abstract.py ├── test_algos.py ├── test_annotations.py ├── test_api.py ├── test_compile.py ├── test_dtype.py ├── test_edge_cases.py ├── test_grad.py ├── test_grad_with_rng.py ├── test_graph_utils.py ├── test_infer.py ├── test_info.py ├── test_lang.py ├── test_macros.py ├── test_model.py ├── test_monomorphize.py ├── test_parser.py ├── test_pipeline.py ├── test_universal.py ├── test_validate.py ├── test_vm.py └── utils ├── __init__.py ├── test_intern.py ├── test_merge.py ├── test_misc.py ├── test_orderedset.py ├── test_partial.py ├── test_trace.py └── test_unify.py /.flake8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/.flake8 -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/.gitignore -------------------------------------------------------------------------------- /.pydocstyle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/.pydocstyle -------------------------------------------------------------------------------- /.travis.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/.travis.yml -------------------------------------------------------------------------------- /ALPHA.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/ALPHA.md -------------------------------------------------------------------------------- /CONTRIBUTING.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/CONTRIBUTING.rst -------------------------------------------------------------------------------- /Jenkinsfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/Jenkinsfile -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/LICENSE -------------------------------------------------------------------------------- /README.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/README.rst -------------------------------------------------------------------------------- /codecov.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/codecov.yml -------------------------------------------------------------------------------- /conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/conftest.py -------------------------------------------------------------------------------- /debug/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/debug/README.md -------------------------------------------------------------------------------- /debug/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/debug/__main__.py -------------------------------------------------------------------------------- /debug/bupdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/debug/bupdb.py -------------------------------------------------------------------------------- /debug/butest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/debug/butest.py -------------------------------------------------------------------------------- /debug/cfg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/debug/cfg.py -------------------------------------------------------------------------------- /debug/cmd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/debug/cmd.py -------------------------------------------------------------------------------- /debug/do_inject.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/debug/do_inject.py -------------------------------------------------------------------------------- /debug/gprint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/debug/gprint.py -------------------------------------------------------------------------------- /debug/graph.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/debug/graph.css -------------------------------------------------------------------------------- /debug/inject.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/debug/inject.py -------------------------------------------------------------------------------- /debug/myia.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/debug/myia.css -------------------------------------------------------------------------------- /debug/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/debug/run.py -------------------------------------------------------------------------------- /debug/steps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/debug/steps.py -------------------------------------------------------------------------------- /debug/test_template.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/debug/test_template.html -------------------------------------------------------------------------------- /debug/tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/debug/tools.py -------------------------------------------------------------------------------- /debug/trace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/debug/trace.py -------------------------------------------------------------------------------- /debug/typ.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/debug/typ.py -------------------------------------------------------------------------------- /devdoc.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/devdoc.md -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/_static/.gitkeep: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/docs/_static/.gitkeep -------------------------------------------------------------------------------- /docs/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/docs/conf.py -------------------------------------------------------------------------------- /docs/gradient.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/docs/gradient.ipynb -------------------------------------------------------------------------------- /docs/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/docs/index.rst -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/environment.yml -------------------------------------------------------------------------------- /examples/lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/examples/lstm.py -------------------------------------------------------------------------------- /examples/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/examples/mlp.py -------------------------------------------------------------------------------- /examples/policy_gradient.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/examples/policy_gradient.py -------------------------------------------------------------------------------- /examples/rnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/examples/rnn.py -------------------------------------------------------------------------------- /examples/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/examples/vae.py -------------------------------------------------------------------------------- /myia/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/__init__.py -------------------------------------------------------------------------------- /myia/abstract/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/abstract/__init__.py -------------------------------------------------------------------------------- /myia/abstract/aliasing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/abstract/aliasing.py -------------------------------------------------------------------------------- /myia/abstract/amerge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/abstract/amerge.py -------------------------------------------------------------------------------- /myia/abstract/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/abstract/data.py -------------------------------------------------------------------------------- /myia/abstract/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/abstract/infer.py -------------------------------------------------------------------------------- /myia/abstract/loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/abstract/loop.py -------------------------------------------------------------------------------- /myia/abstract/macro.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/abstract/macro.py -------------------------------------------------------------------------------- /myia/abstract/ref.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/abstract/ref.py -------------------------------------------------------------------------------- /myia/abstract/to_abstract.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/abstract/to_abstract.py -------------------------------------------------------------------------------- /myia/abstract/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/abstract/utils.py -------------------------------------------------------------------------------- /myia/api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/api.py -------------------------------------------------------------------------------- /myia/classes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/classes.py -------------------------------------------------------------------------------- /myia/compile/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/compile/__init__.py -------------------------------------------------------------------------------- /myia/compile/backends/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/compile/backends/__init__.py -------------------------------------------------------------------------------- /myia/compile/backends/prim_groups.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/compile/backends/prim_groups.py -------------------------------------------------------------------------------- /myia/compile/backends/python/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/compile/backends/python/__init__.py -------------------------------------------------------------------------------- /myia/compile/backends/python/implementations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/compile/backends/python/implementations.py -------------------------------------------------------------------------------- /myia/compile/backends/python/python.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/compile/backends/python/python.py -------------------------------------------------------------------------------- /myia/compile/cconv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/compile/cconv.py -------------------------------------------------------------------------------- /myia/compile/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/compile/transform.py -------------------------------------------------------------------------------- /myia/compile/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/compile/utils.py -------------------------------------------------------------------------------- /myia/compile/vm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/compile/vm.py -------------------------------------------------------------------------------- /myia/debug/finite_diff.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/debug/finite_diff.py -------------------------------------------------------------------------------- /myia/debug/label.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/debug/label.py -------------------------------------------------------------------------------- /myia/debug/traceback.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/debug/traceback.py -------------------------------------------------------------------------------- /myia/debug/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/debug/utils.py -------------------------------------------------------------------------------- /myia/frontends/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/frontends/__init__.py -------------------------------------------------------------------------------- /myia/frontends/abstract_types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/frontends/abstract_types.py -------------------------------------------------------------------------------- /myia/grad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/grad.py -------------------------------------------------------------------------------- /myia/graph_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/graph_utils.py -------------------------------------------------------------------------------- /myia/hypermap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/hypermap.py -------------------------------------------------------------------------------- /myia/info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/info.py -------------------------------------------------------------------------------- /myia/ir/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/ir/__init__.py -------------------------------------------------------------------------------- /myia/ir/anf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/ir/anf.py -------------------------------------------------------------------------------- /myia/ir/clone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/ir/clone.py -------------------------------------------------------------------------------- /myia/ir/manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/ir/manager.py -------------------------------------------------------------------------------- /myia/ir/metagraph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/ir/metagraph.py -------------------------------------------------------------------------------- /myia/ir/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/ir/utils.py -------------------------------------------------------------------------------- /myia/lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/lib.py -------------------------------------------------------------------------------- /myia/monomorphize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/monomorphize.py -------------------------------------------------------------------------------- /myia/operations/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/__init__.py -------------------------------------------------------------------------------- /myia/operations/gen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/gen.py -------------------------------------------------------------------------------- /myia/operations/macro_abstract_array.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/macro_abstract_array.py -------------------------------------------------------------------------------- /myia/operations/macro_apply.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/macro_apply.py -------------------------------------------------------------------------------- /myia/operations/macro_array_len.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/macro_array_len.py -------------------------------------------------------------------------------- /myia/operations/macro_call_object.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/macro_call_object.py -------------------------------------------------------------------------------- /myia/operations/macro_conv2d_grad_input.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/macro_conv2d_grad_input.py -------------------------------------------------------------------------------- /myia/operations/macro_dict_values.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/macro_dict_values.py -------------------------------------------------------------------------------- /myia/operations/macro_dtype.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/macro_dtype.py -------------------------------------------------------------------------------- /myia/operations/macro_einsum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/macro_einsum.py -------------------------------------------------------------------------------- /myia/operations/macro_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/macro_embed.py -------------------------------------------------------------------------------- /myia/operations/macro_getattr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/macro_getattr.py -------------------------------------------------------------------------------- /myia/operations/macro_grad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/macro_grad.py -------------------------------------------------------------------------------- /myia/operations/macro_hasattr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/macro_hasattr.py -------------------------------------------------------------------------------- /myia/operations/macro_is.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/macro_is.py -------------------------------------------------------------------------------- /myia/operations/macro_isinstance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/macro_isinstance.py -------------------------------------------------------------------------------- /myia/operations/macro_make_list.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/macro_make_list.py -------------------------------------------------------------------------------- /myia/operations/macro_resolve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/macro_resolve.py -------------------------------------------------------------------------------- /myia/operations/macro_to_scalar_type.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/macro_to_scalar_type.py -------------------------------------------------------------------------------- /myia/operations/macro_tuple_len.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/macro_tuple_len.py -------------------------------------------------------------------------------- /myia/operations/macro_typeof.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/macro_typeof.py -------------------------------------------------------------------------------- /myia/operations/macro_universal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/macro_universal.py -------------------------------------------------------------------------------- /myia/operations/macro_user_switch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/macro_user_switch.py -------------------------------------------------------------------------------- /myia/operations/op_array_getitem_wrap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/op_array_getitem_wrap.py -------------------------------------------------------------------------------- /myia/operations/op_array_reduce_dim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/op_array_reduce_dim.py -------------------------------------------------------------------------------- /myia/operations/op_enumerate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/op_enumerate.py -------------------------------------------------------------------------------- /myia/operations/op_full.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/op_full.py -------------------------------------------------------------------------------- /myia/operations/op_gadd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/op_gadd.py -------------------------------------------------------------------------------- /myia/operations/op_hyper_map.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/op_hyper_map.py -------------------------------------------------------------------------------- /myia/operations/op_is_not.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/op_is_not.py -------------------------------------------------------------------------------- /myia/operations/op_range.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/op_range.py -------------------------------------------------------------------------------- /myia/operations/op_slice.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/op_slice.py -------------------------------------------------------------------------------- /myia/operations/op_value_and_grad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/op_value_and_grad.py -------------------------------------------------------------------------------- /myia/operations/op_zeros_like.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/op_zeros_like.py -------------------------------------------------------------------------------- /myia/operations/op_zip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/op_zip.py -------------------------------------------------------------------------------- /myia/operations/ops_array.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/ops_array.py -------------------------------------------------------------------------------- /myia/operations/ops_bool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/ops_bool.py -------------------------------------------------------------------------------- /myia/operations/ops_dunder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/ops_dunder.py -------------------------------------------------------------------------------- /myia/operations/ops_nil.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/ops_nil.py -------------------------------------------------------------------------------- /myia/operations/ops_scalar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/ops_scalar.py -------------------------------------------------------------------------------- /myia/operations/ops_string.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/ops_string.py -------------------------------------------------------------------------------- /myia/operations/ops_tuple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/ops_tuple.py -------------------------------------------------------------------------------- /myia/operations/ops_universe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/ops_universe.py -------------------------------------------------------------------------------- /myia/operations/prim_J.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_J.py -------------------------------------------------------------------------------- /myia/operations/prim_Jinv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_Jinv.py -------------------------------------------------------------------------------- /myia/operations/prim_argmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_argmax.py -------------------------------------------------------------------------------- /myia/operations/prim_array_cast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_array_cast.py -------------------------------------------------------------------------------- /myia/operations/prim_array_getitem.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_array_getitem.py -------------------------------------------------------------------------------- /myia/operations/prim_array_map.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_array_map.py -------------------------------------------------------------------------------- /myia/operations/prim_array_max.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_array_max.py -------------------------------------------------------------------------------- /myia/operations/prim_array_reduce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_array_reduce.py -------------------------------------------------------------------------------- /myia/operations/prim_array_scan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_array_scan.py -------------------------------------------------------------------------------- /myia/operations/prim_array_setitem.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_array_setitem.py -------------------------------------------------------------------------------- /myia/operations/prim_array_to_scalar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_array_to_scalar.py -------------------------------------------------------------------------------- /myia/operations/prim_bool_and.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_bool_and.py -------------------------------------------------------------------------------- /myia/operations/prim_bool_eq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_bool_eq.py -------------------------------------------------------------------------------- /myia/operations/prim_bool_not.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_bool_not.py -------------------------------------------------------------------------------- /myia/operations/prim_bool_or.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_bool_or.py -------------------------------------------------------------------------------- /myia/operations/prim_broadcast_shape.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_broadcast_shape.py -------------------------------------------------------------------------------- /myia/operations/prim_casttag.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_casttag.py -------------------------------------------------------------------------------- /myia/operations/prim_concat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_concat.py -------------------------------------------------------------------------------- /myia/operations/prim_conv2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_conv2d.py -------------------------------------------------------------------------------- /myia/operations/prim_conv2d_weight_grad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_conv2d_weight_grad.py -------------------------------------------------------------------------------- /myia/operations/prim_conv_transpose2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_conv_transpose2d.py -------------------------------------------------------------------------------- /myia/operations/prim_dict_getitem.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_dict_getitem.py -------------------------------------------------------------------------------- /myia/operations/prim_dict_setitem.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_dict_setitem.py -------------------------------------------------------------------------------- /myia/operations/prim_distribute.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_distribute.py -------------------------------------------------------------------------------- /myia/operations/prim_dot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_dot.py -------------------------------------------------------------------------------- /myia/operations/prim_env_add.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_env_add.py -------------------------------------------------------------------------------- /myia/operations/prim_env_getitem.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_env_getitem.py -------------------------------------------------------------------------------- /myia/operations/prim_env_setitem.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_env_setitem.py -------------------------------------------------------------------------------- /myia/operations/prim_extract_kwarg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_extract_kwarg.py -------------------------------------------------------------------------------- /myia/operations/prim_gather.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_gather.py -------------------------------------------------------------------------------- /myia/operations/prim_hastag.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_hastag.py -------------------------------------------------------------------------------- /myia/operations/prim_hastype.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_hastype.py -------------------------------------------------------------------------------- /myia/operations/prim_identity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_identity.py -------------------------------------------------------------------------------- /myia/operations/prim_invert_permutation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_invert_permutation.py -------------------------------------------------------------------------------- /myia/operations/prim_make_dict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_make_dict.py -------------------------------------------------------------------------------- /myia/operations/prim_make_exception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_make_exception.py -------------------------------------------------------------------------------- /myia/operations/prim_make_handle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_make_handle.py -------------------------------------------------------------------------------- /myia/operations/prim_make_kwarg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_make_kwarg.py -------------------------------------------------------------------------------- /myia/operations/prim_make_record.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_make_record.py -------------------------------------------------------------------------------- /myia/operations/prim_make_tuple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_make_tuple.py -------------------------------------------------------------------------------- /myia/operations/prim_max_pool2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_max_pool2d.py -------------------------------------------------------------------------------- /myia/operations/prim_max_pool2d_grad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_max_pool2d_grad.py -------------------------------------------------------------------------------- /myia/operations/prim_partial.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_partial.py -------------------------------------------------------------------------------- /myia/operations/prim_raise_.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_raise_.py -------------------------------------------------------------------------------- /myia/operations/prim_random_initialize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_random_initialize.py -------------------------------------------------------------------------------- /myia/operations/prim_random_uint32.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_random_uint32.py -------------------------------------------------------------------------------- /myia/operations/prim_record_getitem.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_record_getitem.py -------------------------------------------------------------------------------- /myia/operations/prim_record_setitem.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_record_setitem.py -------------------------------------------------------------------------------- /myia/operations/prim_reshape.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_reshape.py -------------------------------------------------------------------------------- /myia/operations/prim_return_.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_return_.py -------------------------------------------------------------------------------- /myia/operations/prim_scalar_abs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_scalar_abs.py -------------------------------------------------------------------------------- /myia/operations/prim_scalar_add.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_scalar_add.py -------------------------------------------------------------------------------- /myia/operations/prim_scalar_bit_and.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_scalar_bit_and.py -------------------------------------------------------------------------------- /myia/operations/prim_scalar_bit_lshift.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_scalar_bit_lshift.py -------------------------------------------------------------------------------- /myia/operations/prim_scalar_bit_not.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_scalar_bit_not.py -------------------------------------------------------------------------------- /myia/operations/prim_scalar_bit_or.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_scalar_bit_or.py -------------------------------------------------------------------------------- /myia/operations/prim_scalar_bit_rshift.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_scalar_bit_rshift.py -------------------------------------------------------------------------------- /myia/operations/prim_scalar_bit_xor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_scalar_bit_xor.py -------------------------------------------------------------------------------- /myia/operations/prim_scalar_cast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_scalar_cast.py -------------------------------------------------------------------------------- /myia/operations/prim_scalar_cos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_scalar_cos.py -------------------------------------------------------------------------------- /myia/operations/prim_scalar_div.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_scalar_div.py -------------------------------------------------------------------------------- /myia/operations/prim_scalar_eq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_scalar_eq.py -------------------------------------------------------------------------------- /myia/operations/prim_scalar_exp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_scalar_exp.py -------------------------------------------------------------------------------- /myia/operations/prim_scalar_floor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_scalar_floor.py -------------------------------------------------------------------------------- /myia/operations/prim_scalar_ge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_scalar_ge.py -------------------------------------------------------------------------------- /myia/operations/prim_scalar_gt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_scalar_gt.py -------------------------------------------------------------------------------- /myia/operations/prim_scalar_le.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_scalar_le.py -------------------------------------------------------------------------------- /myia/operations/prim_scalar_log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_scalar_log.py -------------------------------------------------------------------------------- /myia/operations/prim_scalar_lt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_scalar_lt.py -------------------------------------------------------------------------------- /myia/operations/prim_scalar_max.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_scalar_max.py -------------------------------------------------------------------------------- /myia/operations/prim_scalar_mod.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_scalar_mod.py -------------------------------------------------------------------------------- /myia/operations/prim_scalar_mul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_scalar_mul.py -------------------------------------------------------------------------------- /myia/operations/prim_scalar_ne.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_scalar_ne.py -------------------------------------------------------------------------------- /myia/operations/prim_scalar_pow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_scalar_pow.py -------------------------------------------------------------------------------- /myia/operations/prim_scalar_sign.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_scalar_sign.py -------------------------------------------------------------------------------- /myia/operations/prim_scalar_sin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_scalar_sin.py -------------------------------------------------------------------------------- /myia/operations/prim_scalar_sub.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_scalar_sub.py -------------------------------------------------------------------------------- /myia/operations/prim_scalar_tan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_scalar_tan.py -------------------------------------------------------------------------------- /myia/operations/prim_scalar_tanh.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_scalar_tanh.py -------------------------------------------------------------------------------- /myia/operations/prim_scalar_to_array.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_scalar_to_array.py -------------------------------------------------------------------------------- /myia/operations/prim_scalar_trunc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_scalar_trunc.py -------------------------------------------------------------------------------- /myia/operations/prim_scalar_uadd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_scalar_uadd.py -------------------------------------------------------------------------------- /myia/operations/prim_scalar_usub.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_scalar_usub.py -------------------------------------------------------------------------------- /myia/operations/prim_scatter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_scatter.py -------------------------------------------------------------------------------- /myia/operations/prim_scatter_add.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_scatter_add.py -------------------------------------------------------------------------------- /myia/operations/prim_shape.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_shape.py -------------------------------------------------------------------------------- /myia/operations/prim_split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_split.py -------------------------------------------------------------------------------- /myia/operations/prim_stop_gradient.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_stop_gradient.py -------------------------------------------------------------------------------- /myia/operations/prim_string_eq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_string_eq.py -------------------------------------------------------------------------------- /myia/operations/prim_switch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_switch.py -------------------------------------------------------------------------------- /myia/operations/prim_tagged.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_tagged.py -------------------------------------------------------------------------------- /myia/operations/prim_take.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_take.py -------------------------------------------------------------------------------- /myia/operations/prim_take_grad_inp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_take_grad_inp.py -------------------------------------------------------------------------------- /myia/operations/prim_transpose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_transpose.py -------------------------------------------------------------------------------- /myia/operations/prim_tuple_getitem.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_tuple_getitem.py -------------------------------------------------------------------------------- /myia/operations/prim_tuple_setitem.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_tuple_setitem.py -------------------------------------------------------------------------------- /myia/operations/prim_universe_getitem.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_universe_getitem.py -------------------------------------------------------------------------------- /myia/operations/prim_universe_setitem.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_universe_setitem.py -------------------------------------------------------------------------------- /myia/operations/prim_unsafe_static_cast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/prim_unsafe_static_cast.py -------------------------------------------------------------------------------- /myia/operations/primitives.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/primitives.py -------------------------------------------------------------------------------- /myia/operations/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/operations/utils.py -------------------------------------------------------------------------------- /myia/opt/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/opt/__init__.py -------------------------------------------------------------------------------- /myia/opt/cse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/opt/cse.py -------------------------------------------------------------------------------- /myia/opt/dde.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/opt/dde.py -------------------------------------------------------------------------------- /myia/opt/lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/opt/lib.py -------------------------------------------------------------------------------- /myia/opt/opt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/opt/opt.py -------------------------------------------------------------------------------- /myia/opt/rewrite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/opt/rewrite.py -------------------------------------------------------------------------------- /myia/parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/parser.py -------------------------------------------------------------------------------- /myia/pipeline/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/pipeline/__init__.py -------------------------------------------------------------------------------- /myia/pipeline/pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/pipeline/pipeline.py -------------------------------------------------------------------------------- /myia/pipeline/resources.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/pipeline/resources.py -------------------------------------------------------------------------------- /myia/pipeline/standard.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/pipeline/standard.py -------------------------------------------------------------------------------- /myia/pipeline/steps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/pipeline/steps.py -------------------------------------------------------------------------------- /myia/public_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/public_api.py -------------------------------------------------------------------------------- /myia/simplify_types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/simplify_types.py -------------------------------------------------------------------------------- /myia/testing/__init__.py: -------------------------------------------------------------------------------- 1 | """Testing utilities, made available for plugins.""" 2 | -------------------------------------------------------------------------------- /myia/testing/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/testing/common.py -------------------------------------------------------------------------------- /myia/testing/multitest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/testing/multitest.py -------------------------------------------------------------------------------- /myia/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/utils/__init__.py -------------------------------------------------------------------------------- /myia/utils/env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/utils/env.py -------------------------------------------------------------------------------- /myia/utils/errors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/utils/errors.py -------------------------------------------------------------------------------- /myia/utils/get_fields.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/utils/get_fields.py -------------------------------------------------------------------------------- /myia/utils/intern.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/utils/intern.py -------------------------------------------------------------------------------- /myia/utils/merge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/utils/merge.py -------------------------------------------------------------------------------- /myia/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/utils/misc.py -------------------------------------------------------------------------------- /myia/utils/orderedset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/utils/orderedset.py -------------------------------------------------------------------------------- /myia/utils/partial.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/utils/partial.py -------------------------------------------------------------------------------- /myia/utils/trace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/utils/trace.py -------------------------------------------------------------------------------- /myia/utils/unify.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/utils/unify.py -------------------------------------------------------------------------------- /myia/utils/universe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/utils/universe.py -------------------------------------------------------------------------------- /myia/utils/variables.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/utils/variables.py -------------------------------------------------------------------------------- /myia/validate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/validate.py -------------------------------------------------------------------------------- /myia/vm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/vm.py -------------------------------------------------------------------------------- /myia/xtype.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia/xtype.py -------------------------------------------------------------------------------- /myia_backend_pytorch/README.rst: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /myia_backend_pytorch/cpu-extras.conda: -------------------------------------------------------------------------------- 1 | pytorch::cpuonly 2 | -------------------------------------------------------------------------------- /myia_backend_pytorch/environment-cpu.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia_backend_pytorch/environment-cpu.yml -------------------------------------------------------------------------------- /myia_backend_pytorch/environment-gpu.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia_backend_pytorch/environment-gpu.yml -------------------------------------------------------------------------------- /myia_backend_pytorch/gpu-extras.conda: -------------------------------------------------------------------------------- 1 | cudatoolkit=10.2.* 2 | cudnn=7.6.* 3 | -------------------------------------------------------------------------------- /myia_backend_pytorch/myia_backend_pytorch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia_backend_pytorch/myia_backend_pytorch/__init__.py -------------------------------------------------------------------------------- /myia_backend_pytorch/myia_backend_pytorch/pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia_backend_pytorch/myia_backend_pytorch/pytorch.py -------------------------------------------------------------------------------- /myia_backend_pytorch/myia_backend_pytorch/pytorch_conv_grad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia_backend_pytorch/myia_backend_pytorch/pytorch_conv_grad.py -------------------------------------------------------------------------------- /myia_backend_pytorch/myia_tests_backend_pytorch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia_backend_pytorch/myia_tests_backend_pytorch/__init__.py -------------------------------------------------------------------------------- /myia_backend_pytorch/poetry.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia_backend_pytorch/poetry.lock -------------------------------------------------------------------------------- /myia_backend_pytorch/pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia_backend_pytorch/pyproject.toml -------------------------------------------------------------------------------- /myia_backend_pytorch/scripts/gen_conda_env_file.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia_backend_pytorch/scripts/gen_conda_env_file.sh -------------------------------------------------------------------------------- /myia_backend_pytorch/tests/test_pytorch_rng.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia_backend_pytorch/tests/test_pytorch_rng.py -------------------------------------------------------------------------------- /myia_backend_relay/README.rst: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /myia_backend_relay/environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia_backend_relay/environment.yml -------------------------------------------------------------------------------- /myia_backend_relay/myia_backend_relay/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia_backend_relay/myia_backend_relay/__init__.py -------------------------------------------------------------------------------- /myia_backend_relay/myia_backend_relay/relay.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia_backend_relay/myia_backend_relay/relay.py -------------------------------------------------------------------------------- /myia_backend_relay/myia_backend_relay/relay_helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia_backend_relay/myia_backend_relay/relay_helpers.py -------------------------------------------------------------------------------- /myia_backend_relay/myia_backend_relay/relay_philox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia_backend_relay/myia_backend_relay/relay_philox.py -------------------------------------------------------------------------------- /myia_backend_relay/myia_tests_backend_relay/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia_backend_relay/myia_tests_backend_relay/__init__.py -------------------------------------------------------------------------------- /myia_backend_relay/poetry.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia_backend_relay/poetry.lock -------------------------------------------------------------------------------- /myia_backend_relay/pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia_backend_relay/pyproject.toml -------------------------------------------------------------------------------- /myia_backend_relay/relay-extras.conda: -------------------------------------------------------------------------------- 1 | mila-iqia::tvm=0.7dev1+5.* 2 | -------------------------------------------------------------------------------- /myia_backend_relay/scripts/gen_conda_env_file.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia_backend_relay/scripts/gen_conda_env_file.sh -------------------------------------------------------------------------------- /myia_backend_relay/tests/compile/test_relay.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia_backend_relay/tests/compile/test_relay.py -------------------------------------------------------------------------------- /myia_backend_relay/tests/test_compile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia_backend_relay/tests/test_compile.py -------------------------------------------------------------------------------- /myia_backend_relay/tests/test_relay_rng.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia_backend_relay/tests/test_relay_rng.py -------------------------------------------------------------------------------- /myia_frontend_pytorch/README.rst: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /myia_frontend_pytorch/cpu-extras.conda: -------------------------------------------------------------------------------- 1 | pytorch::cpuonly 2 | -------------------------------------------------------------------------------- /myia_frontend_pytorch/environment-cpu.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia_frontend_pytorch/environment-cpu.yml -------------------------------------------------------------------------------- /myia_frontend_pytorch/environment-gpu.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia_frontend_pytorch/environment-gpu.yml -------------------------------------------------------------------------------- /myia_frontend_pytorch/gpu-extras.conda: -------------------------------------------------------------------------------- 1 | cudatoolkit=10.2.* 2 | cudnn=7.6.* 3 | -------------------------------------------------------------------------------- /myia_frontend_pytorch/myia_frontend_pytorch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia_frontend_pytorch/myia_frontend_pytorch/__init__.py -------------------------------------------------------------------------------- /myia_frontend_pytorch/myia_frontend_pytorch/pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia_frontend_pytorch/myia_frontend_pytorch/pytorch.py -------------------------------------------------------------------------------- /myia_frontend_pytorch/myia_frontend_pytorch/pytorch_abstract_types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia_frontend_pytorch/myia_frontend_pytorch/pytorch_abstract_types.py -------------------------------------------------------------------------------- /myia_frontend_pytorch/poetry.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia_frontend_pytorch/poetry.lock -------------------------------------------------------------------------------- /myia_frontend_pytorch/pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia_frontend_pytorch/pyproject.toml -------------------------------------------------------------------------------- /myia_frontend_pytorch/scripts/gen_conda_env_file.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia_frontend_pytorch/scripts/gen_conda_env_file.sh -------------------------------------------------------------------------------- /myia_frontend_pytorch/tests/test_validate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia_frontend_pytorch/tests/test_validate.py -------------------------------------------------------------------------------- /myia_utils/README.rst: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /myia_utils/myia_utils/__init__.py: -------------------------------------------------------------------------------- 1 | """Helper scripts for myia installation.""" 2 | -------------------------------------------------------------------------------- /myia_utils/myia_utils/update_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia_utils/myia_utils/update_env.py -------------------------------------------------------------------------------- /myia_utils/poetry.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia_utils/poetry.lock -------------------------------------------------------------------------------- /myia_utils/pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/myia_utils/pyproject.toml -------------------------------------------------------------------------------- /poetry.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/poetry.lock -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/pyproject.toml -------------------------------------------------------------------------------- /pytest.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/pytest.ini -------------------------------------------------------------------------------- /scripts/build-doc.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/scripts/build-doc.sh -------------------------------------------------------------------------------- /scripts/check.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/scripts/check.sh -------------------------------------------------------------------------------- /scripts/ci-install.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/scripts/ci-install.sh -------------------------------------------------------------------------------- /scripts/composite.py.template: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/scripts/composite.py.template -------------------------------------------------------------------------------- /scripts/consolidate-coverage.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/scripts/consolidate-coverage.sh -------------------------------------------------------------------------------- /scripts/gen_conda_env_file.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/scripts/gen_conda_env_file.sh -------------------------------------------------------------------------------- /scripts/infprim.py.template: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/scripts/infprim.py.template -------------------------------------------------------------------------------- /scripts/macro.py.template: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/scripts/macro.py.template -------------------------------------------------------------------------------- /scripts/new_operation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/scripts/new_operation.py -------------------------------------------------------------------------------- /scripts/op.py.template: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/scripts/op.py.template -------------------------------------------------------------------------------- /scripts/prim.py.template: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/scripts/prim.py.template -------------------------------------------------------------------------------- /scripts/regen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/scripts/regen.py -------------------------------------------------------------------------------- /scripts/test_op.py.template: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/scripts/test_op.py.template -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/compile/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/compile/test_backend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/compile/test_backend.py -------------------------------------------------------------------------------- /tests/compile/test_cconv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/compile/test_cconv.py -------------------------------------------------------------------------------- /tests/composite/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/composite/examples/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/composite/examples/operations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/composite/examples/operations.py -------------------------------------------------------------------------------- /tests/composite/examples/prim_composite_full.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/composite/examples/prim_composite_full.py -------------------------------------------------------------------------------- /tests/composite/examples/prim_composite_simple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/composite/examples/prim_composite_simple.py -------------------------------------------------------------------------------- /tests/composite/examples/primitives.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/composite/examples/primitives.py -------------------------------------------------------------------------------- /tests/composite/test_composite_examples.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/composite/test_composite_examples.py -------------------------------------------------------------------------------- /tests/debug/test_debug_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/debug/test_debug_utils.py -------------------------------------------------------------------------------- /tests/frontends/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/frontends/test_all_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/frontends/test_all_ops.py -------------------------------------------------------------------------------- /tests/frontends/test_frontend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/frontends/test_frontend.py -------------------------------------------------------------------------------- /tests/frontends/test_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/frontends/test_pytorch.py -------------------------------------------------------------------------------- /tests/ir/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/ir/test_anf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/ir/test_anf.py -------------------------------------------------------------------------------- /tests/ir/test_clone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/ir/test_clone.py -------------------------------------------------------------------------------- /tests/ir/test_ir_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/ir/test_ir_utils.py -------------------------------------------------------------------------------- /tests/ir/test_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/ir/test_manager.py -------------------------------------------------------------------------------- /tests/operations/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/operations/test_hyper_map.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/operations/test_hyper_map.py -------------------------------------------------------------------------------- /tests/operations/test_macro_dtype.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/operations/test_macro_dtype.py -------------------------------------------------------------------------------- /tests/operations/test_macro_einsum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/operations/test_macro_einsum.py -------------------------------------------------------------------------------- /tests/operations/test_operations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/operations/test_operations.py -------------------------------------------------------------------------------- /tests/operations/test_ops_dunder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/operations/test_ops_dunder.py -------------------------------------------------------------------------------- /tests/operations/test_raise_assert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/operations/test_raise_assert.py -------------------------------------------------------------------------------- /tests/operations/test_universe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/operations/test_universe.py -------------------------------------------------------------------------------- /tests/opt/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/opt/test_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/opt/test_lib.py -------------------------------------------------------------------------------- /tests/opt/test_opt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/opt/test_opt.py -------------------------------------------------------------------------------- /tests/opt/test_rewrite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/opt/test_rewrite.py -------------------------------------------------------------------------------- /tests/prim/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/prim/test_py_implementations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/prim/test_py_implementations.py -------------------------------------------------------------------------------- /tests/python_backend/.pdbrc: -------------------------------------------------------------------------------- 1 | next 2 | list . 3 | continue 4 | quit 5 | -------------------------------------------------------------------------------- /tests/python_backend/test_frontend_compilation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/python_backend/test_frontend_compilation.py -------------------------------------------------------------------------------- /tests/python_backend/test_graph_compilation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/python_backend/test_graph_compilation.py -------------------------------------------------------------------------------- /tests/python_backend/test_python_debug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/python_backend/test_python_debug.py -------------------------------------------------------------------------------- /tests/python_backend/test_python_pdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/python_backend/test_python_pdb.py -------------------------------------------------------------------------------- /tests/python_backend/test_python_rng.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/python_backend/test_python_rng.py -------------------------------------------------------------------------------- /tests/test_abstract.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/test_abstract.py -------------------------------------------------------------------------------- /tests/test_algos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/test_algos.py -------------------------------------------------------------------------------- /tests/test_annotations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/test_annotations.py -------------------------------------------------------------------------------- /tests/test_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/test_api.py -------------------------------------------------------------------------------- /tests/test_compile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/test_compile.py -------------------------------------------------------------------------------- /tests/test_dtype.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/test_dtype.py -------------------------------------------------------------------------------- /tests/test_edge_cases.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/test_edge_cases.py -------------------------------------------------------------------------------- /tests/test_grad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/test_grad.py -------------------------------------------------------------------------------- /tests/test_grad_with_rng.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/test_grad_with_rng.py -------------------------------------------------------------------------------- /tests/test_graph_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/test_graph_utils.py -------------------------------------------------------------------------------- /tests/test_infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/test_infer.py -------------------------------------------------------------------------------- /tests/test_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/test_info.py -------------------------------------------------------------------------------- /tests/test_lang.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/test_lang.py -------------------------------------------------------------------------------- /tests/test_macros.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/test_macros.py -------------------------------------------------------------------------------- /tests/test_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/test_model.py -------------------------------------------------------------------------------- /tests/test_monomorphize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/test_monomorphize.py -------------------------------------------------------------------------------- /tests/test_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/test_parser.py -------------------------------------------------------------------------------- /tests/test_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/test_pipeline.py -------------------------------------------------------------------------------- /tests/test_universal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/test_universal.py -------------------------------------------------------------------------------- /tests/test_validate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/test_validate.py -------------------------------------------------------------------------------- /tests/test_vm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/test_vm.py -------------------------------------------------------------------------------- /tests/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/utils/test_intern.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/utils/test_intern.py -------------------------------------------------------------------------------- /tests/utils/test_merge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/utils/test_merge.py -------------------------------------------------------------------------------- /tests/utils/test_misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/utils/test_misc.py -------------------------------------------------------------------------------- /tests/utils/test_orderedset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/utils/test_orderedset.py -------------------------------------------------------------------------------- /tests/utils/test_partial.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/utils/test_partial.py -------------------------------------------------------------------------------- /tests/utils/test_trace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/utils/test_trace.py -------------------------------------------------------------------------------- /tests/utils/test_unify.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mila-iqia/myia/HEAD/tests/utils/test_unify.py --------------------------------------------------------------------------------