Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Does slangtorch support uint64_t scalar type in TorchTensor? #26

Open
ozchango opened this issue Dec 17, 2024 · 2 comments
Open

Does slangtorch support uint64_t scalar type in TorchTensor? #26

ozchango opened this issue Dec 17, 2024 · 2 comments

Comments

@ozchango
Copy link

ozchango commented Dec 17, 2024

Thank you for the awesome work. I want to allocate a new uint64_t type tensor.

uint P = conic.size(0);
TorchTensor<uint64_t3> dL_duvd_int = TorchTensor<uint64_t3>.alloc(P, 3);
TorchTensor<uint64_t3> dL_dconic_int = TorchTensor<uint64_t3>.alloc(P, 3);
TorchTensor<uint64_t> dL_dopacity_int = TorchTensor<uint64_t>.alloc(P);
TorchTensor<uint64_t3> dL_dfeature_int = TorchTensor<uint64_t3>.alloc(P, 3);
TorchTensor<uint64_t2> dL_dndc_int = TorchTensor<uint64_t2>.alloc(P, 2);

However, it seems that the scalar type is not supported.
image

I am using CUDA 12.2, pytorch 2.1.1 and slangtorch 1.2.6

@bmillsNV
Copy link

Thanks for raising this. Is this issue blocking you? We plan to get to this one in Q1 2025. Any concerns with that timeline?

@bmillsNV
Copy link

bmillsNV commented Jan 9, 2025

No feedback, so planed for this quarter.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

2 participants