-
Notifications
You must be signed in to change notification settings - Fork 25.7k
Open
Labels
good first issuemodule: cudaRelated to torch.cuda, and CUDA support in generalRelated to torch.cuda, and CUDA support in generalmodule: jiteratortriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate moduleThis issue has been looked at a team member, and triaged and prioritized into an appropriate module
Description
Issue description
The following complex kernels cause CI failures when implemented with jitted_gpu_kernel.
Some may simply require a small tolerance increase to the tests but for larger differences we need to figure out which implementation is correct, thrust or jitterator.
Note that gpu_kernel implemenations use thrust while jitted_gpu_kernel calls the complex math implementations we vendor from libc++. These can be found in llvm_complex.cpp.
cc @ngimel @mruberry @ezyang @soumith @msaroufim @wconstab @bdhirsh @anijain2305
Metadata
Metadata
Assignees
Labels
good first issuemodule: cudaRelated to torch.cuda, and CUDA support in generalRelated to torch.cuda, and CUDA support in generalmodule: jiteratortriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate moduleThis issue has been looked at a team member, and triaged and prioritized into an appropriate module