Hugging Face's logo

kernels-community
/
moe

kernel
Model card Files Community
1
danieldk HF staff commited on
Commit
3d21f81
·
1 Parent(s): c2ee5f1

Normalize some directory names

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. build.toml +4 -4
  2. {ext-torch/moe/utils → tests}/__init__.py +0 -0
  3. {test → tests/kernels}/__init__.py +0 -0
  4. {test → tests}/kernels/test_moe.py +0 -0
  5. {test → tests}/kernels/utils.py +0 -0
  6. {ext-torch → torch-ext}/moe/__init__.py +0 -0
  7. {ext-torch → torch-ext}/moe/configs/E=1,N=14336,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a16.json +0 -0
  8. {ext-torch → torch-ext}/moe/configs/E=1,N=14336,device_name=NVIDIA_A100-SXM4-80GB.json +0 -0
  9. {ext-torch → torch-ext}/moe/configs/E=1,N=1792,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a16.json +0 -0
  10. {ext-torch → torch-ext}/moe/configs/E=1,N=1792,device_name=NVIDIA_A100-SXM4-80GB.json +0 -0
  11. {ext-torch → torch-ext}/moe/configs/E=1,N=3072,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a16.json +0 -0
  12. {ext-torch → torch-ext}/moe/configs/E=1,N=3072,device_name=NVIDIA_H100_80GB_HBM3,dtype=int8_w8a16.json +0 -0
  13. {ext-torch → torch-ext}/moe/configs/E=1,N=3072,device_name=NVIDIA_H100_80GB_HBM3.json +0 -0
  14. {ext-torch → torch-ext}/moe/configs/E=1,N=3584,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a16.json +0 -0
  15. {ext-torch → torch-ext}/moe/configs/E=1,N=3584,device_name=NVIDIA_A100-SXM4-80GB.json +0 -0
  16. {ext-torch → torch-ext}/moe/configs/E=1,N=7168,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a16.json +0 -0
  17. {ext-torch → torch-ext}/moe/configs/E=1,N=7168,device_name=NVIDIA_A100-SXM4-80GB.json +0 -0
  18. {ext-torch → torch-ext}/moe/configs/E=16,N=1344,device_name=NVIDIA_A100-SXM4-40GB.json +0 -0
  19. {ext-torch → torch-ext}/moe/configs/E=16,N=1344,device_name=NVIDIA_A100-SXM4-80GB.json +0 -0
  20. {ext-torch → torch-ext}/moe/configs/E=16,N=1344,device_name=NVIDIA_H100_80GB_HBM3.json +0 -0
  21. {ext-torch → torch-ext}/moe/configs/E=16,N=14336,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a16.json +0 -0
  22. {ext-torch → torch-ext}/moe/configs/E=16,N=14336,device_name=NVIDIA_A100-SXM4-80GB.json +0 -0
  23. {ext-torch → torch-ext}/moe/configs/E=16,N=1792,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a16.json +0 -0
  24. {ext-torch → torch-ext}/moe/configs/E=16,N=1792,device_name=NVIDIA_A100-SXM4-80GB.json +0 -0
  25. {ext-torch → torch-ext}/moe/configs/E=16,N=2688,device_name=NVIDIA_A100-SXM4-80GB.json +0 -0
  26. {ext-torch → torch-ext}/moe/configs/E=16,N=2688,device_name=NVIDIA_H100_80GB_HBM3.json +0 -0
  27. {ext-torch → torch-ext}/moe/configs/E=16,N=3072,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a16.json +0 -0
  28. {ext-torch → torch-ext}/moe/configs/E=16,N=3072,device_name=NVIDIA_H100_80GB_HBM3,dtype=int8_w8a16.json +0 -0
  29. {ext-torch → torch-ext}/moe/configs/E=16,N=3200,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8.json +0 -0
  30. {ext-torch → torch-ext}/moe/configs/E=16,N=3584,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a16.json +0 -0
  31. {ext-torch → torch-ext}/moe/configs/E=16,N=3584,device_name=NVIDIA_A100-SXM4-80GB.json +0 -0
  32. {ext-torch → torch-ext}/moe/configs/E=16,N=6400,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8.json +0 -0
  33. {ext-torch → torch-ext}/moe/configs/E=16,N=7168,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a16.json +0 -0
  34. {ext-torch → torch-ext}/moe/configs/E=16,N=7168,device_name=NVIDIA_A100-SXM4-80GB.json +0 -0
  35. {ext-torch → torch-ext}/moe/configs/E=16,N=7168,device_name=NVIDIA_H100_80GB_HBM3,dtype=int8_w8a16.json +0 -0
  36. {ext-torch → torch-ext}/moe/configs/E=16,N=800,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8.json +0 -0
  37. {ext-torch → torch-ext}/moe/configs/E=64,N=1280,device_name=NVIDIA_A100-SXM4-80GB.json +0 -0
  38. {ext-torch → torch-ext}/moe/configs/E=64,N=1280,device_name=NVIDIA_H100_80GB_HBM3.json +0 -0
  39. {ext-torch → torch-ext}/moe/configs/E=64,N=640,device_name=NVIDIA_A100-SXM4-80GB.json +0 -0
  40. {ext-torch → torch-ext}/moe/configs/E=64,N=640,device_name=NVIDIA_H100_80GB_HBM3.json +0 -0
  41. {ext-torch → torch-ext}/moe/configs/E=8,N=14336,device_name=AMD_Instinct_MI300X.json +0 -0
  42. {ext-torch → torch-ext}/moe/configs/E=8,N=14336,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8.json +0 -0
  43. {ext-torch → torch-ext}/moe/configs/E=8,N=1792,device_name=AMD_Instinct_MI300X.json +0 -0
  44. {ext-torch → torch-ext}/moe/configs/E=8,N=1792,device_name=NVIDIA_A100-SXM4-40GB.json +0 -0
  45. {ext-torch → torch-ext}/moe/configs/E=8,N=1792,device_name=NVIDIA_A100-SXM4-80GB.json +0 -0
  46. {ext-torch → torch-ext}/moe/configs/E=8,N=1792,device_name=NVIDIA_H100_80GB_HBM3.json +0 -0
  47. {ext-torch → torch-ext}/moe/configs/E=8,N=2048,device_name=NVIDIA_A100-SXM4-80GB.json +0 -0
  48. {ext-torch → torch-ext}/moe/configs/E=8,N=2048,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8.json +0 -0
  49. {ext-torch → torch-ext}/moe/configs/E=8,N=2048,device_name=NVIDIA_H100_80GB_HBM3.json +0 -0
  50. {ext-torch → torch-ext}/moe/configs/E=8,N=3584,device_name=AMD_Instinct_MI300X.json +0 -0
build.toml CHANGED
@@ -5,12 +5,12 @@ version = "0.0.1"
5
  name = "moe"
6
  src = [
7
  "core/scalar_type.hpp",
8
- "ext-torch/registration.h",
9
- "ext-torch/torch_binding.cpp",
10
- "ext-torch/torch_binding.h",
11
  ]
12
  include = ["."]
13
- pyroot = "ext-torch"
14
  pyext = ["py", "json"]
15
 
16
  [kernel.fp8]
 
5
  name = "moe"
6
  src = [
7
  "core/scalar_type.hpp",
8
+ "torch-ext/registration.h",
9
+ "torch-ext/torch_binding.cpp",
10
+ "torch-ext/torch_binding.h",
11
  ]
12
  include = ["."]
13
+ pyroot = "torch-ext"
14
  pyext = ["py", "json"]
15
 
16
  [kernel.fp8]
{ext-torch/moe/utils → tests}/__init__.py RENAMED
File without changes
{test → tests/kernels}/__init__.py RENAMED
File without changes
{test → tests}/kernels/test_moe.py RENAMED
File without changes
{test → tests}/kernels/utils.py RENAMED
File without changes
{ext-torch → torch-ext}/moe/__init__.py RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=1,N=14336,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a16.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=1,N=14336,device_name=NVIDIA_A100-SXM4-80GB.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=1,N=1792,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a16.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=1,N=1792,device_name=NVIDIA_A100-SXM4-80GB.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=1,N=3072,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a16.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=1,N=3072,device_name=NVIDIA_H100_80GB_HBM3,dtype=int8_w8a16.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=1,N=3072,device_name=NVIDIA_H100_80GB_HBM3.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=1,N=3584,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a16.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=1,N=3584,device_name=NVIDIA_A100-SXM4-80GB.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=1,N=7168,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a16.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=1,N=7168,device_name=NVIDIA_A100-SXM4-80GB.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=16,N=1344,device_name=NVIDIA_A100-SXM4-40GB.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=16,N=1344,device_name=NVIDIA_A100-SXM4-80GB.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=16,N=1344,device_name=NVIDIA_H100_80GB_HBM3.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=16,N=14336,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a16.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=16,N=14336,device_name=NVIDIA_A100-SXM4-80GB.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=16,N=1792,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a16.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=16,N=1792,device_name=NVIDIA_A100-SXM4-80GB.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=16,N=2688,device_name=NVIDIA_A100-SXM4-80GB.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=16,N=2688,device_name=NVIDIA_H100_80GB_HBM3.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=16,N=3072,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a16.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=16,N=3072,device_name=NVIDIA_H100_80GB_HBM3,dtype=int8_w8a16.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=16,N=3200,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=16,N=3584,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a16.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=16,N=3584,device_name=NVIDIA_A100-SXM4-80GB.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=16,N=6400,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=16,N=7168,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a16.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=16,N=7168,device_name=NVIDIA_A100-SXM4-80GB.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=16,N=7168,device_name=NVIDIA_H100_80GB_HBM3,dtype=int8_w8a16.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=16,N=800,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=64,N=1280,device_name=NVIDIA_A100-SXM4-80GB.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=64,N=1280,device_name=NVIDIA_H100_80GB_HBM3.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=64,N=640,device_name=NVIDIA_A100-SXM4-80GB.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=64,N=640,device_name=NVIDIA_H100_80GB_HBM3.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=8,N=14336,device_name=AMD_Instinct_MI300X.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=8,N=14336,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=8,N=1792,device_name=AMD_Instinct_MI300X.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=8,N=1792,device_name=NVIDIA_A100-SXM4-40GB.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=8,N=1792,device_name=NVIDIA_A100-SXM4-80GB.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=8,N=1792,device_name=NVIDIA_H100_80GB_HBM3.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=8,N=2048,device_name=NVIDIA_A100-SXM4-80GB.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=8,N=2048,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=8,N=2048,device_name=NVIDIA_H100_80GB_HBM3.json RENAMED
File without changes
{ext-torch → torch-ext}/moe/configs/E=8,N=3584,device_name=AMD_Instinct_MI300X.json RENAMED
File without changes