PR #7219 Register custom call with CUDA plugin

Register custom call with CUDA plugin #7219

bhavya01 merged 8 commits into master from triton_plugin2

688bb72d

Remove --gpu flag while build torch cuda

4e9d1f1a

Remove check GPU step

48c9552a

Update TORCH_CUDA_ARCH_LIST

cb65fe2a

bhavya01 marked this pull request as ready for review 1 year ago

Fix runner for _test_requiring_torch_cuda

29852a8c

Run python tests for multiples GPUs only

893c7022

bhavya01 requested a review from

will-cromar 1 year ago

Remove print statements

b3fd65ec

bhavya01 requested a review from

vanbasten23 1 year ago

bhavya01 requested a review from

alanwaketan 1 year ago

bhavya01 assigned

bhavya01 1 year ago

bhavya01 added Triton

vanbasten23 commented on 2024-06-11

Add openxla reference for registering custom call with plugin

4489cea3

bhavya01 requested a review from

vanbasten23 1 year ago

vanbasten23 approved these changes on 2024-06-12

alanwaketan approved these changes on 2024-06-12

bhavya01 merged 86d8967c into master 1 year ago

Reviewers

alanwaketan

vanbasten23

will-cromar

Assignees

bhavya01

Labels

Triton

Milestone

No milestone