JuliaPy / JuliaPy/PythonCall.jl

parse gpu array from python to Julia

Đang mở
#93 7 bình luận 0 reaction 0 người được giao Xem trên GitHub
Ngôn ngữ chính
Julia
Star
1.1k
Fork
86
Merge trung bình
1 ngày 22 giờ
Pull request đã merge (30 ngày)
3

Mô tả

hello I have cupy cuda array and I want to pass it into julia as is.
CUDA arrays are just list of pointers so it should be possible
from CUDA.jl side I know is possible as I have a comment
"""
for passing data the other way around you can use unsafe_wrap(CuArray, ...) to create a CUDA.jl array from a device pointer you get from Python
"""
still I can not make it work - anybody have some working example?

What I was trying
```
import cupy
import numba
import numpy as np
import torch
import torch.utils.dlpack
from statistics import median
import timeit
from juliacall import Main as jl
# julia.install()
from numba import cuda
jl.seval("using Pkg")
jl.seval("""Pkg.add("CUDA")""")
jl.seval("""Pkg.add("PythonCall")""")
jl.seval("""using CUDA""")
jl.seval("""using PythonCall""")
jl.seval("""CUDA.allowscalar(true)""")
jl.seval("""print(sum(CUDA.ones(3,3,3)))""")# working good
jl.seval("""function bb(arrGold)
# print(CUDA.unsafe_wrap(CuArray{UInt8,3},arrGold, (2,2,2)))
print( pyconvert(CuArray{UInt8} ,arrGold ))
end""")

def print_hi(name):
t1 = torch.cuda.ByteTensor(np.ones((2,2,2)))
c1 = cupy.asarray(t1)

Main.bb(c1)

def forBenchPymia():
numba.cuda.synchronize()
jl.bb(c1)
numba.cuda.synchronize()

num_runs = 1
num_repetions = 1#2
ex_time = timeit.Timer(forBenchPymia).repeat(
repeat=num_repetions,
number=num_runs)
res= median(ex_time)*1000
print("bench")
print(res)

if __name__ == '__main__':
print_hi('PyCharm')

```

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

Chưa lập chỉ mục được hướng dẫn đóng góp cho kho mã nguồn này

Đánh giá

Issue này chưa được đánh giá.

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.