CIL icon indicating copy to clipboard operation
CIL copied to clipboard

Switch to ASTRA direct_FP3D/BP3D interface

Open wjp opened this issue 1 year ago • 0 comments

Description

Switch the Astra wrapper to use Astra's direct_FP3D and direct_BP3D functions for 3D Astra geometries. These functions provide a more concise interface and avoid unnecessary data copies.

This interface is supported from Astra 2.0. Starting from Astra 2.3, these functions also transparently accept DLPack tensors (including torch cpu/cuda tensors).

Example Usage

All unit tests pass for me. This is a result from the CIL PyTorch integration hackathon, and has therefore only seen limited further testing.

wjp avatar Apr 09 '25 08:04 wjp