mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
30cb6ac53c
Summary:
Apple recently announced ML Compute, a new framework available in macOS Big Sur, which enables users to accelerate the training of neural networks on Mac hardware. This PR is the first on a series of PRs that will enable the integration with ML Compute. Most of the integration code will live on a separate subrepo named `mlc`.
The integration with `mlc` (ML Compute) will be very similar to that of xla. We rely on registering our ops through:
TORCH_LIBRARY_IMPL(aten, PrivateUse1, m) {
m.impl_UNBOXED(<op_schema_name>, &customized_op_kernel)
...
}
Pull Request resolved: https://github.com/pytorch/pytorch/pull/50634
Reviewed By: malfet
Differential Revision: D26614213
Pulled By: smessmer
fbshipit-source-id: 3b492b346c61cc3950ac880ac01a82fbdddbc07b
|
||
|---|---|---|
| .. | ||
| auto_gil.h | ||
| byte_order.cpp | ||
| byte_order.h | ||
| cuda_enabled.h | ||
| cuda_lazy_init.cpp | ||
| cuda_lazy_init.h | ||
| disable_torch_function.cpp | ||
| disable_torch_function.h | ||
| disallow_copy.h | ||
| init.cpp | ||
| init.h | ||
| invalid_arguments.cpp | ||
| invalid_arguments.h | ||
| memory.h | ||
| numpy_stub.h | ||
| object_ptr.cpp | ||
| object_ptr.h | ||
| out_types.cpp | ||
| out_types.h | ||
| pybind.h | ||
| pycfunction_helpers.h | ||
| python_arg_parser.cpp | ||
| python_arg_parser.h | ||
| python_compat.h | ||
| python_dispatch.cpp | ||
| python_dispatch.h | ||
| python_numbers.h | ||
| python_scalars.h | ||
| python_strings.h | ||
| python_stub.h | ||
| python_tuples.h | ||
| six.h | ||
| structseq.cpp | ||
| structseq.h | ||
| tensor_apply.cpp | ||
| tensor_apply.h | ||
| tensor_dtypes.cpp | ||
| tensor_dtypes.h | ||
| tensor_flatten.cpp | ||
| tensor_flatten.h | ||
| tensor_layouts.cpp | ||
| tensor_layouts.h | ||
| tensor_list.cpp | ||
| tensor_list.h | ||
| tensor_memoryformats.cpp | ||
| tensor_memoryformats.h | ||
| tensor_new.cpp | ||
| tensor_new.h | ||
| tensor_numpy.cpp | ||
| tensor_numpy.h | ||
| tensor_qschemes.cpp | ||
| tensor_qschemes.h | ||
| tensor_types.cpp | ||
| tensor_types.h | ||
| throughput_benchmark-inl.h | ||
| throughput_benchmark.cpp | ||
| throughput_benchmark.h | ||
| variadic.cpp | ||
| variadic.h | ||