icecuber commited on
Commit
aa68bc1
·
verified ·
1 Parent(s): a6a3aa2

Upload build/torch-universal/kernel_test/__init__.py with huggingface_hub

Browse files
build/torch-universal/kernel_test/__init__.py ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ import torch
2
+
3
+ def mul2(x: torch.Tensor) -> torch.Tensor:
4
+ return x*2
5
+
6
+ __all__ = ["mul2"]