Skip to content

GitLab

Commit 38c8b3ac authored by rusty1s's avatar rusty1s
Browse files

cuda kernel

parent a2f18da3
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 8 deletions
torch_scatter/utils/ffi.py deleted 100644 → 0
View file @ a2f18da3
from .._ext import ffi
def get_func(name, tensor):
name += '_'
name += 'cuda_' if tensor.is_cuda else ''
name += tensor.type().split('.')[-1][:-6]
return getattr(ffi, name)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment