xla icon indicating copy to clipboard operation
xla copied to clipboard

Merge C++ and Mesh implementation of most Mesh methods.

Open copybara-service[bot] opened this issue 2 years ago • 0 comments

Merge C++ and Mesh implementation of most Mesh methods.

C++ becomes the source of truth for Mesh.

changes are in layout.py, tensor_layout.[cc|h], and the pywrap_dtensor_device.cc file.

Many attribute methods are converted to properties. If users are broken, will use forward fix.

I'll update Layout after this stablizes.

copybara-service[bot] avatar Mar 09 '23 18:03 copybara-service[bot]