()
| 1082 | |
| 1083 | |
| 1084 | def get_packages(): |
| 1085 | _packages = [ |
| 1086 | "triton", |
| 1087 | "triton/_C", |
| 1088 | "triton/backends", |
| 1089 | "triton/compiler", |
| 1090 | "triton/language", |
| 1091 | "triton/language/extra", |
| 1092 | "triton/runtime", |
| 1093 | "triton/tools", |
| 1094 | "triton/tools/extra", |
| 1095 | ] |
| 1096 | # Dynamically enumerate triton/experimental subpackages |
| 1097 | _packages += get_triton_subpackages("triton/experimental", "triton/experimental") |
| 1098 | # Dynamically enumerate triton/extension subpackages (for Ascend) |
| 1099 | if _is_ascend_platform() and check_env_flag("TRITON_USE_ASCEND", "OFF"): |
| 1100 | _packages += get_triton_subpackages("triton/extension", "triton/extension") |
| 1101 | _packages += [f'triton/backends/{backend.name}' for backend in backends] |
| 1102 | _packages += get_extra_packages("language") |
| 1103 | _packages += get_extra_packages("tools") |
| 1104 | if check_env_flag("TRITON_BUILD_PROTON", "ON"): # Default ON |
| 1105 | _packages += ["triton/profiler"] |
| 1106 | if check_env_flag("TRITON_BUILD_DISTRIBUTED", "ON"): # Default ON |
| 1107 | _packages += [ |
| 1108 | "triton_dist", |
| 1109 | "triton_dist/_C", |
| 1110 | "triton_dist/benchmark", |
| 1111 | "triton_dist/kernels", |
| 1112 | "triton_dist/kernels/nvidia", |
| 1113 | "triton_dist/kernels/amd", |
| 1114 | "triton_dist/kernels/metax", |
| 1115 | "triton_dist/language", |
| 1116 | "triton_dist/language/extra", |
| 1117 | "triton_dist/language/extra/cuda", |
| 1118 | "triton_dist/language/extra/hip", |
| 1119 | "triton_dist/language/extra/ascend", |
| 1120 | "triton_dist/language/extra/maca", |
| 1121 | "triton_dist/mega_triton_kernel", |
| 1122 | "triton_dist/function", |
| 1123 | "triton_dist/function/nvidia", |
| 1124 | "triton_dist/function/amd", |
| 1125 | "triton_dist/layers", |
| 1126 | "triton_dist/layers/nvidia", |
| 1127 | "triton_dist/layers/amd", |
| 1128 | "triton_dist/models", |
| 1129 | "triton_dist/test", |
| 1130 | "triton_dist/test/ascend", |
| 1131 | "triton_dist/tools", |
| 1132 | "triton_dist/tools/compile", |
| 1133 | "triton_dist/tools/profiler", |
| 1134 | "triton_dist/tools/runtime", |
| 1135 | "triton_dist/tools/tune", |
| 1136 | ] |
| 1137 | if _is_maca_platform() and check_env_flag("TRITON_USE_MACA", "OFF"): |
| 1138 | _packages += ["triton/pymxshmem", "triton_dist/language/extra/maca"] |
| 1139 | |
| 1140 | if check_env_flag("TRITON_BUILD_LITTLE_KERNEL", "ON"): # Default ON |
| 1141 | _packages += [ |
no test coverage detected