Skip to content

[Minor] Handle the Device Index

Benjamin Halimi requested to merge cuda_device into dev

Description

Current state of the code doesn't take care of the device index. This MR proposes to fix this.

Files modified : Tensor.hpp, Tensor.cpp, pybind_Tensor.cpp

Merge request reports

Copyright © Eclipse Foundation, Inc. All Rights Reserved.     Privacy Policy | Terms of Use | Copyright Agent