[Draft] Experimental mdarray

[fbusato]

Description

Goal: Provide a synchronous memory-owning version of mdspan for both host and device memory.

• The functionalities should be aligned with thrust::host/device_vector to not surprise users.
• Unlike thrust::host/device_vector, mdarray is NOT a container and doesn't support iterators.
• Should be able to replace Rapids implementation.

The implementation is based on the CRTP pattern + mdspan inheritance. The host/device_mdarray class are minimal.
It supports the standard copy/move semantic and host/device interaction, e.g. copy device to host

Open points:

• asynchronous version. Possible but could be misused. Out-of-scope for the first PR.

Todo list:

• [Most Important] get feedback
• Mirror device_mdarray test for host_mdarray
• Consider non-initialized version of constructors.
• Address TODO in the code.
• Documentation.

[copy-pr-bot]

Auto-sync is disabled for draft pull requests in this repository. Workflows must be run manually.

Contributors can view more details about this message here.

[github-actions]

😬 CI Workflow Results

🟥 Finished in 1h 31m: Pass: 72%/122 | Total: 2d 03h | Max: 1h 30m | Hits: 93%/196064

See results here.