diff --git a/CMakeLists.txt b/CMakeLists.txt index f05a53604c0d..b38066e59f3d 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -515,3 +515,7 @@ install(DIRECTORY ${PROJECT_SOURCE_DIR}/include/ COMPONENT headers DESTINATION ${CMAKE_INSTALL_INCLUDEDIR} FILES_MATCHING PATTERN "*.td") +install(DIRECTORY ${PROJECT_SOURCE_DIR}/python/ + COMPONENT headers + DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}/python + FILES_MATCHING PATTERN "*.h") diff --git a/bin/RegisterTritonDialects.h b/bin/RegisterTritonDialects.h index 02b5e575b27c..0bcb32e58091 100644 --- a/bin/RegisterTritonDialects.h +++ b/bin/RegisterTritonDialects.h @@ -46,7 +46,7 @@ #include "mlir/Conversion/UBToLLVM/UBToLLVM.h" #include "triton/Tools/PluginUtils.h" -#include "triton/Tools/Sys/GetEnv.hpp" +#include "triton/Tools/Sys/GetEnv.h" namespace mlir { namespace test { diff --git a/include/triton/Tools/Sys/Dump.hpp b/include/triton/Tools/Sys/Dump.h similarity index 83% rename from include/triton/Tools/Sys/Dump.hpp rename to include/triton/Tools/Sys/Dump.h index 95f8ee95caa8..52d061af07fb 100644 --- a/include/triton/Tools/Sys/Dump.hpp +++ b/include/triton/Tools/Sys/Dump.h @@ -1,7 +1,7 @@ -#ifndef TRITON_TOOLS_SYS_DUMP_HPP -#define TRITON_TOOLS_SYS_DUMP_HPP +#ifndef TRITON_TOOLS_SYS_DUMP_H +#define TRITON_TOOLS_SYS_DUMP_H -#include "triton/Tools/Sys/GetEnv.hpp" +#include "triton/Tools/Sys/GetEnv.h" #include "llvm/Support/FileSystem.h" #include "llvm/Support/raw_ostream.h" diff --git a/include/triton/Tools/Sys/GetEnv.hpp b/include/triton/Tools/Sys/GetEnv.h similarity index 95% rename from include/triton/Tools/Sys/GetEnv.hpp rename to include/triton/Tools/Sys/GetEnv.h index 8c0d8611cf2f..d9fc2b5fcfb7 100644 --- a/include/triton/Tools/Sys/GetEnv.hpp +++ b/include/triton/Tools/Sys/GetEnv.h @@ -1,5 +1,5 @@ -#ifndef TRITON_TOOLS_SYS_GETENV_HPP -#define TRITON_TOOLS_SYS_GETENV_HPP +#ifndef TRITON_TOOLS_SYS_GETENV_H +#define TRITON_TOOLS_SYS_GETENV_H #include #include @@ -69,7 +69,7 @@ inline void assertIsRecognized(const std::string &env) { bool is_neutral = CACHE_NEUTRAL_ENV_VARS.find(env.c_str()) != CACHE_NEUTRAL_ENV_VARS.end(); std::string errmsg = env + "is not recognized. " - "Please add it to triton/tools/sys/getenv.hpp"; + "Please add it to triton/Tools/Sys/GetEnv.h"; assert((is_invalidating || is_neutral) && errmsg.c_str()); } diff --git a/lib/Analysis/Utility.cpp b/lib/Analysis/Utility.cpp index dda93e16627d..3133797d6e42 100644 --- a/lib/Analysis/Utility.cpp +++ b/lib/Analysis/Utility.cpp @@ -17,7 +17,7 @@ #include "triton/Dialect/TritonGPU/IR/LinearLayoutConversions.h" #include "triton/Tools/LayoutUtils.h" #include "triton/Tools/LinearLayout.h" -#include "triton/Tools/Sys/GetEnv.hpp" +#include "triton/Tools/Sys/GetEnv.h" #include "llvm/ADT/SmallSet.h" namespace mlir { diff --git a/lib/Dialect/TritonGPU/Transforms/Pipeliner/AssignLatencies.cpp b/lib/Dialect/TritonGPU/Transforms/Pipeliner/AssignLatencies.cpp index b977ac969005..302dfc4c0e42 100644 --- a/lib/Dialect/TritonGPU/Transforms/Pipeliner/AssignLatencies.cpp +++ b/lib/Dialect/TritonGPU/Transforms/Pipeliner/AssignLatencies.cpp @@ -6,7 +6,7 @@ #include "triton/Dialect/TritonGPU/Transforms/Schedule.h" #include "triton/Dialect/TritonGPU/Transforms/Utility.h" #include "triton/Dialect/TritonNvidiaGPU/IR/Dialect.h" -#include "triton/Tools/Sys/GetEnv.hpp" +#include "triton/Tools/Sys/GetEnv.h" #include "llvm/Support/Debug.h" #define DEBUG_TYPE "triton-loop-pipeline" diff --git a/lib/Dialect/TritonGPU/Transforms/Pipeliner/SoftwarePipeliner.cpp b/lib/Dialect/TritonGPU/Transforms/Pipeliner/SoftwarePipeliner.cpp index 9005277d124b..ca7613789273 100644 --- a/lib/Dialect/TritonGPU/Transforms/Pipeliner/SoftwarePipeliner.cpp +++ b/lib/Dialect/TritonGPU/Transforms/Pipeliner/SoftwarePipeliner.cpp @@ -16,7 +16,7 @@ #include "triton/Dialect/TritonGPU/Transforms/Schedule.h" #include "triton/Dialect/TritonGPU/Transforms/Utility.h" #include "triton/Dialect/TritonNvidiaGPU/IR/Dialect.h" -#include "triton/Tools/Sys/Dump.hpp" +#include "triton/Tools/Sys/Dump.h" //===----------------------------------------------------------------------===// // This file will create a schedule that will be handed over to the pipeline // expander. diff --git a/lib/Dialect/TritonGPU/Transforms/WarpSpecialization/PartitionSchedulingUtility.cpp b/lib/Dialect/TritonGPU/Transforms/WarpSpecialization/PartitionSchedulingUtility.cpp index d7defc0932da..10e4fafcc8f6 100644 --- a/lib/Dialect/TritonGPU/Transforms/WarpSpecialization/PartitionSchedulingUtility.cpp +++ b/lib/Dialect/TritonGPU/Transforms/WarpSpecialization/PartitionSchedulingUtility.cpp @@ -1,6 +1,6 @@ #include "triton/Dialect/TritonGPU/Transforms/PartitionSchedulingUtility.h" #include "mlir/Support/LLVM.h" -#include "triton/Tools/Sys/GetEnv.hpp" +#include "triton/Tools/Sys/GetEnv.h" #include #include diff --git a/lib/Dialect/TritonNvidiaGPU/IR/Dialect.cpp b/lib/Dialect/TritonNvidiaGPU/IR/Dialect.cpp index c7d4ce6677be..e40a1bcd2c3f 100644 --- a/lib/Dialect/TritonNvidiaGPU/IR/Dialect.cpp +++ b/lib/Dialect/TritonNvidiaGPU/IR/Dialect.cpp @@ -24,7 +24,7 @@ #include "triton/Dialect/Triton/IR/Dialect.h" #include "triton/Dialect/Triton/IR/Utility.h" #include "triton/Dialect/TritonGPU/IR/TritonGPUInterfaces.h" -#include "triton/Tools/Sys/GetEnv.hpp" +#include "triton/Tools/Sys/GetEnv.h" #include diff --git a/lib/Dialect/TritonNvidiaGPU/Transforms/FenceInsertion.cpp b/lib/Dialect/TritonNvidiaGPU/Transforms/FenceInsertion.cpp index f831ed3a9bab..6efb52c49cd4 100644 --- a/lib/Dialect/TritonNvidiaGPU/Transforms/FenceInsertion.cpp +++ b/lib/Dialect/TritonNvidiaGPU/Transforms/FenceInsertion.cpp @@ -3,7 +3,7 @@ #include "triton/Dialect/TritonGPU/Transforms/Utility.h" #include "triton/Dialect/TritonNvidiaGPU/IR/Dialect.h" #include "triton/Dialect/TritonNvidiaGPU/Transforms/Passes.h" -#include "triton/Tools/Sys/GetEnv.hpp" +#include "triton/Tools/Sys/GetEnv.h" #include "llvm/Support/Debug.h" //===----------------------------------------------------------------------===// diff --git a/lib/Dialect/TritonNvidiaGPU/Transforms/PromoteLHSToTMem.cpp b/lib/Dialect/TritonNvidiaGPU/Transforms/PromoteLHSToTMem.cpp index a3107f6da53b..d70404529b2b 100644 --- a/lib/Dialect/TritonNvidiaGPU/Transforms/PromoteLHSToTMem.cpp +++ b/lib/Dialect/TritonNvidiaGPU/Transforms/PromoteLHSToTMem.cpp @@ -6,7 +6,7 @@ #include "triton/Dialect/TritonGPU/Transforms/Utility.h" #include "triton/Dialect/TritonNvidiaGPU/IR/Dialect.h" #include "triton/Dialect/TritonNvidiaGPU/Transforms/Passes.h" -#include "triton/Tools/Sys/GetEnv.hpp" +#include "triton/Tools/Sys/GetEnv.h" namespace ttg = mlir::triton::gpu; diff --git a/lib/Target/LLVMIR/LLVMDIScope.cpp b/lib/Target/LLVMIR/LLVMDIScope.cpp index 0595673ff2ed..a6bb9cd9b784 100644 --- a/lib/Target/LLVMIR/LLVMDIScope.cpp +++ b/lib/Target/LLVMIR/LLVMDIScope.cpp @@ -4,7 +4,7 @@ #include "mlir/Pass/Pass.h" #include "mlir/Support/LLVM.h" #include "triton/Target/LLVMIR/Passes.h" -#include "triton/Tools/Sys/GetEnv.hpp" +#include "triton/Tools/Sys/GetEnv.h" #include "llvm/BinaryFormat/Dwarf.h" #include "llvm/Support/Debug.h" #include "llvm/Support/Path.h" diff --git a/lib/Tools/PluginUtils.cpp b/lib/Tools/PluginUtils.cpp index 05bbcc89fb09..5dc41b263b86 100644 --- a/lib/Tools/PluginUtils.cpp +++ b/lib/Tools/PluginUtils.cpp @@ -1,5 +1,5 @@ #include "triton/Tools/PluginUtils.h" -#include "triton/Tools/Sys/GetEnv.hpp" +#include "triton/Tools/Sys/GetEnv.h" #include "llvm/Support/Debug.h" #include "llvm/Support/Error.h" diff --git a/python/src/ir.cc b/python/src/ir.cc index 02763e59c38f..6a9c9baba888 100644 --- a/python/src/ir.cc +++ b/python/src/ir.cc @@ -37,8 +37,8 @@ #include "triton/Dialect/TritonNvidiaGPU/IR/Dialect.h" #include "triton/Dialect/TritonNvidiaGPU/Transforms/TMAUtilities.h" #include "triton/Tools/PluginUtils.h" -#include "triton/Tools/Sys/Dump.hpp" -#include "triton/Tools/Sys/GetEnv.hpp" +#include "triton/Tools/Sys/Dump.h" +#include "triton/Tools/Sys/GetEnv.h" #include "llvm/Support/SourceMgr.h" namespace { diff --git a/python/src/ir.h b/python/src/ir.h index f8dd9b2941ac..f001bfbf1237 100644 --- a/python/src/ir.h +++ b/python/src/ir.h @@ -1,7 +1,7 @@ #pragma once #include "mlir/IR/Builders.h" #include "mlir/IR/DialectRegistry.h" -#include "triton/Tools/Sys/GetEnv.hpp" +#include "triton/Tools/Sys/GetEnv.h" #include #include diff --git a/python/src/llvm.cc b/python/src/llvm.cc index 759a9b3c235b..4fa590db6355 100644 --- a/python/src/llvm.cc +++ b/python/src/llvm.cc @@ -1,7 +1,7 @@ #include "mlir/IR/BuiltinOps.h" // mlir::ModuleOp #include "mlir/Target/LLVMIR/LLVMTranslationInterface.h" #include "mlir/Target/LLVMIR/ModuleTranslation.h" -#include "triton/Tools/Sys/GetEnv.hpp" +#include "triton/Tools/Sys/GetEnv.h" #include "llvm/ADT/SmallVector.h" #include "llvm/Analysis/ScopedNoAliasAA.h" #include "llvm/CodeGen/MIRParser/MIRParser.h" diff --git a/python/src/passes.cc b/python/src/passes.cc index 9f10eebaa99e..8fc6f2388984 100644 --- a/python/src/passes.cc +++ b/python/src/passes.cc @@ -13,7 +13,7 @@ #include "triton/Dialect/TritonInstrument/Transforms/Passes.h" #include "triton/Target/LLVMIR/Passes.h" #include "triton/Tools/PluginUtils.h" -#include "triton/Tools/Sys/GetEnv.hpp" +#include "triton/Tools/Sys/GetEnv.h" #include #include #include diff --git a/third_party/amd/lib/TritonAMDGPUToLLVM/MaskedOpsToLLVM.cpp b/third_party/amd/lib/TritonAMDGPUToLLVM/MaskedOpsToLLVM.cpp index bd90cc52c3db..7b99ffa87efd 100644 --- a/third_party/amd/lib/TritonAMDGPUToLLVM/MaskedOpsToLLVM.cpp +++ b/third_party/amd/lib/TritonAMDGPUToLLVM/MaskedOpsToLLVM.cpp @@ -10,7 +10,7 @@ #include "mlir/Pass/Pass.h" #include "mlir/Transforms/GreedyPatternRewriteDriver.h" #include "triton/Dialect/TritonGPU/Transforms/Utility.h" -#include "triton/Tools/Sys/GetEnv.hpp" +#include "triton/Tools/Sys/GetEnv.h" #include using namespace mlir; diff --git a/third_party/nvidia/hopper/lib/Transforms/WarpSpecialization.cpp b/third_party/nvidia/hopper/lib/Transforms/WarpSpecialization.cpp index c0dbcdb9b48a..15ce4ca8c7ee 100644 --- a/third_party/nvidia/hopper/lib/Transforms/WarpSpecialization.cpp +++ b/third_party/nvidia/hopper/lib/Transforms/WarpSpecialization.cpp @@ -7,7 +7,7 @@ #include "nvidia/include/Dialect/NVWS/IR/Dialect.h" #include "triton/Dialect/TritonGPU/IR/Dialect.h" #include "triton/Dialect/TritonGPU/Transforms/PipeliningUtility.h" -#include "triton/Tools/Sys/Dump.hpp" +#include "triton/Tools/Sys/Dump.h" #define DEBUG_TYPE "nvgpu-warp-specialization" #define DBGS() (llvm::dbgs() << "[" DEBUG_TYPE "]: ") diff --git a/third_party/nvidia/hopper/lib/Transforms/WarpSpecialization/WSBuffer.cpp b/third_party/nvidia/hopper/lib/Transforms/WarpSpecialization/WSBuffer.cpp index d04aead8b914..a8dd013cada2 100644 --- a/third_party/nvidia/hopper/lib/Transforms/WarpSpecialization/WSBuffer.cpp +++ b/third_party/nvidia/hopper/lib/Transforms/WarpSpecialization/WSBuffer.cpp @@ -21,7 +21,7 @@ #include "triton/Dialect/TritonGPU/Transforms/Passes.h" #include "triton/Dialect/TritonGPU/Transforms/PipeliningUtility.h" #include "triton/Dialect/TritonGPU/Transforms/TritonGPUConversion.h" -#include "triton/Tools/Sys/GetEnv.hpp" +#include "triton/Tools/Sys/GetEnv.h" #include #include diff --git a/third_party/nvidia/hopper/lib/Transforms/WarpSpecialization/WSLowerMem.cpp b/third_party/nvidia/hopper/lib/Transforms/WarpSpecialization/WSLowerMem.cpp index 92ab99caa2eb..52b2eabc99ff 100644 --- a/third_party/nvidia/hopper/lib/Transforms/WarpSpecialization/WSLowerMem.cpp +++ b/third_party/nvidia/hopper/lib/Transforms/WarpSpecialization/WSLowerMem.cpp @@ -21,7 +21,7 @@ #include "triton/Dialect/TritonGPU/Transforms/Passes.h" #include "triton/Dialect/TritonGPU/Transforms/PipeliningUtility.h" #include "triton/Dialect/TritonGPU/Transforms/TritonGPUConversion.h" -#include "triton/Tools/Sys/GetEnv.hpp" +#include "triton/Tools/Sys/GetEnv.h" #include #include diff --git a/third_party/nvidia/hopper/lib/Transforms/WarpSpecialization/WSSpecialize.cpp b/third_party/nvidia/hopper/lib/Transforms/WarpSpecialization/WSSpecialize.cpp index 02ae2b05dc82..6c3f1b342d81 100644 --- a/third_party/nvidia/hopper/lib/Transforms/WarpSpecialization/WSSpecialize.cpp +++ b/third_party/nvidia/hopper/lib/Transforms/WarpSpecialization/WSSpecialize.cpp @@ -22,7 +22,7 @@ #include "triton/Dialect/TritonGPU/Transforms/Passes.h" #include "triton/Dialect/TritonGPU/Transforms/PipeliningUtility.h" #include "triton/Dialect/TritonGPU/Transforms/TritonGPUConversion.h" -#include "triton/Tools/Sys/GetEnv.hpp" +#include "triton/Tools/Sys/GetEnv.h" #include #include