xp.eye(..., device=device) has not effect when xp is cupy

Đang mở
#337 3 bình luận 0 reaction 0 người được giao Xem trên GitHub

Chưa có ai nhận issue này.

Đánh giá

Độ khó
3/5
Thời gian dự kiến
1-2 ngày
Mức phù hợp với người mới
38/100
Loại issue
Lỗi
Độ rõ ràng
Khá rõ ràng
Mức độ hoạt động
Đình trệ
Công nghệ
python
Lĩnh vực
data

Hướng nghiên cứu

Bắt đầu bằng cách chạy trình tái hiện được cung cấp cho CuPy và PyTorch với các phiên bản được liệt kê, sau đó lần theo array_api_compat.array_namespace(x) và lệnh gọi xp.eye để so sánh cách xử lý thiết bị. Được xem là hoàn tất khi namespace của CuPy tôn trọng device=1 như trường hợp PyTorch, với một bài kiểm thử hồi quy bao quát hành vi này.

Do mô hình lập chỉ mục viết ra từ nội dung của issue.

Mô tả

blocked by upstream

Dear all,

I run the following command when xp is cupy

# cupy
x = cupy.asarray([1.0] * 3)
xp = array_api_compat.array_namespace(x)
E = xp.eye(3, 3, k=0, dtype='float64', device=1)
print(E.device)
with xp.cuda.Device(1):
    E = xp.eye(3, 3, k=0, dtype='float64', device=1)
    print(E.device)
# torch
x = torch.as_tensor([1.0] * 3)
xp = array_api_compat.array_namespace(x)
E = xp.eye(3, 3, k=0, dtype=torch.float64, device='cuda:1')
print(E.device)

and I get the result:

<CUDA Device 0>
<CUDA Device 1>
cuda:1

It seems that device argument of xp.eye has no effect here when xp is cupy.
It seems to work well when xp is torch.
Did I miss something?

Thank you.

array-api-compat 1.12.0
cupy-cuda12x 13.4.1

Ngôn ngữ chính
Python
Star
131
Fork
49
Merge trung bình
2 ngày 8 giờ
Pull request đã merge (30 ngày)
6

Hướng dẫn đóng góp

Mở hướng dẫn đóng góp

Bắt đầu từ đâu

  1. Đọc hết issue, rồi đọc hướng dẫn đóng góp của dự án.
  2. Bình luận trên issue rằng bạn sẽ nhận — tránh hai người làm cùng một việc.
  3. Fork repository và làm thay đổi trên một nhánh.
  4. Mở pull request có tham chiếu số hiệu của issue.

Issue khác của data-apis/array-api-compat

Tất cả issue của data-apis/array-api-compat

Issue tương tự

Thêm issue về Python

Nhận issue mới trong hộp thư của bạn

Bản tóm tắt ngắn những issue GitHub phù hợp với người mới.