xla
[backport] Register custom call with CUDA plugin
#7303
Merged
Overview
Commits
8
Changes
View On GitHub

[backport] Register custom call with CUDA plugin #7303

bhavya01 merged 8 commits into r2.4 from triton_plugin2
bhavya01
bhavya01 Register custom call with CUDA plugin
688bb72d
bhavya01 Remove --gpu flag while build torch cuda
4e9d1f1a
bhavya01 Remove check GPU step
48c9552a
bhavya01 Update TORCH_CUDA_ARCH_LIST
cb65fe2a
bhavya01 Fix runner for _test_requiring_torch_cuda
29852a8c
bhavya01 Run python tests for multiples GPUs only
893c7022
bhavya01 Remove print statements
b3fd65ec
bhavya01 Add openxla reference for registering custom call with plugin
4489cea3
bhavya01 bhavya01 requested a review from lsy323 lsy323 1 year ago
bhavya01 bhavya01 changed the title Register custom call with CUDA plugin [backport] Register custom call with CUDA plugin 1 year ago
bhavya01 bhavya01 requested a review from vanbasten23 vanbasten23 1 year ago
vanbasten23
vanbasten23 approved these changes on 2024-06-24
bhavya01 bhavya01 merged e44787a5 into r2.4 1 year ago
miladm miladm added Triton

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Triton
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum