├── .clang-format ├── .editorconfig ├── .git-blame-ignore-revs ├── .github ├── CODEOWNERS ├── ISSUE_TEMPLATE │ ├── bug.yml │ ├── config.yml │ └── performance.yml ├── PULL_REQUEST_TEMPLATE.md ├── dependabot.yml └── workflows │ ├── build-macos.yml │ ├── ci.yml │ ├── create_release.yml │ ├── documentation.yml │ ├── llvm-build.yml │ ├── llvm-build │ ├── almalinux.Dockerfile │ └── centos.Dockerfile │ ├── pre-commit.yml │ ├── runner-preparation.yml │ └── wheels.yml ├── .gitignore ├── .pre-commit-config.yaml ├── CMakeLists.txt ├── CONTRIBUTING.md ├── LICENSE ├── MANIFEST.in ├── Makefile ├── README.md ├── RELEASE.md ├── bin ├── CMakeLists.txt ├── RegisterTritonDialects.h ├── triton-llvm-opt.cpp ├── triton-lsp.cpp ├── triton-opt.cpp ├── triton-reduce.cpp └── triton-tensor-layout.cpp ├── cmake ├── AddTritonUnitTest.cmake ├── FindLLVM.cmake ├── json-version.txt ├── llvm-hash.txt └── nvidia-toolchain-version.json ├── docs ├── Makefile ├── _templates │ └── versions.html ├── backend │ ├── ldmatrixOperand0.svg │ └── ldmatrixOperand1.svg ├── conf.py ├── getting-started │ ├── installation.rst │ └── tutorials │ │ ├── grouped_vs_row_major_ordering.png │ │ ├── parallel_reduction.png │ │ └── random_bits.png ├── index.rst ├── meetups │ ├── 01-24-2024 │ │ └── notes.md │ ├── 02-20-2024 │ │ ├── Proton.pdf │ │ └── notes.md │ ├── 03-12-2025 │ │ └── notes.md │ ├── 04-02-2024 │ │ └── notes.md │ ├── 05-01-2025 │ │ └── notes.md │ ├── 05-07-2024 │ │ └── notes.md │ ├── 07-09-2025 │ │ └── notes.md │ ├── 07-18-2023 │ │ └── notes.md │ ├── 08-06-2024 │ │ └── notes.md │ ├── 08-22-2023 │ │ ├── amd-update.pdf │ │ ├── intel-xpu-update.pptx │ │ └── notes.md │ ├── 10-25-2023 │ │ ├── intel-xpu-update.pdf │ │ ├── notes.md │ │ └── triton-shared.pptx │ ├── 12-13-2023 │ │ └── notes.md │ ├── dev-meetup-2023.md │ ├── dev_conference_2024.md │ └── for_moderators │ │ └── README.md ├── programming-guide │ ├── chapter-1 │ │ ├── cuda-parallel-matmul.png │ │ ├── introduction.rst │ │ └── triton-parallel-matmul.png │ ├── chapter-2 │ │ ├── halide-iteration.png │ │ ├── polyhedral-iteration.png │ │ └── related-work.rst │ └── chapter-3 │ │ └── debugging.rst ├── python-api │ ├── triton-semantics.rst │ ├── triton.language.extra.cuda.rst │ ├── triton.language.rst │ ├── triton.rst │ └── triton.testing.rst └── requirements.txt ├── include ├── CMakeLists.txt └── triton │ ├── Analysis │ ├── Alias.h │ ├── Allocation.h │ ├── AxisInfo.h │ ├── Membar.h │ └── Utility.h │ ├── CMakeLists.txt │ ├── Conversion │ ├── CMakeLists.txt │ ├── MLIRTypes.h │ ├── TritonGPUToLLVM │ │ ├── AllocateSharedMemoryUtility.h │ │ ├── AsmFormat.h │ │ ├── CMakeLists.txt │ │ ├── ElementwiseOpToLLVMBase.h │ │ ├── FMADotUtility.h │ │ ├── Passes.h │ │ ├── Passes.td │ │ ├── PatternTritonGPUOpToLLVM.h │ │ ├── TargetInfoBase.h │ │ ├── TypeConverter.h │ │ └── Utility.h │ └── TritonToTritonGPU │ │ ├── CMakeLists.txt │ │ ├── Passes.h │ │ └── Passes.td │ ├── Dialect │ ├── CMakeLists.txt │ ├── Gluon │ │ ├── CMakeCache.txt │ │ ├── CMakeLists.txt │ │ ├── IR │ │ │ ├── CMakeLists.txt │ │ │ ├── Dialect.h │ │ │ ├── GluonAttrDefs.td │ │ │ ├── GluonDialect.td │ │ │ └── GluonOps.td │ │ └── Transforms │ │ │ ├── CMakeLists.txt │ │ │ ├── Passes.h │ │ │ └── Passes.td │ ├── Triton │ │ ├── CMakeLists.txt │ │ ├── IR │ │ │ ├── CMakeLists.txt │ │ │ ├── Dialect.h │ │ │ ├── DiscardableAttributes.h │ │ │ ├── Interfaces.h │ │ │ ├── OpInterfaces.h │ │ │ ├── Traits.h │ │ │ ├── TritonAttrDefs.td │ │ │ ├── TritonDialect.td │ │ │ ├── TritonInterfaces.td │ │ │ ├── TritonOpInterfaces.td │ │ │ ├── TritonOps.td │ │ │ ├── TritonTypes.td │ │ │ ├── Types.h │ │ │ └── Utility.h │ │ └── Transforms │ │ │ ├── ArithTypeConversion.h │ │ │ ├── CMakeLists.txt │ │ │ ├── FunctionTypeConversion.h │ │ │ ├── LoopPeeling.h │ │ │ ├── Passes.h │ │ │ └── Passes.td │ ├── TritonGPU │ │ ├── CMakeLists.txt │ │ ├── IR │ │ │ ├── Attributes.h │ │ │ ├── CMakeLists.txt │ │ │ ├── Dialect.h │ │ │ ├── LayoutUtility.h │ │ │ ├── LinearLayoutConversions.h │ │ │ ├── Traits.h │ │ │ ├── TritonGPUAttrDefs.td │ │ │ ├── TritonGPUDialect.td │ │ │ ├── TritonGPUInterfaces.h │ │ │ ├── TritonGPUOps.td │ │ │ ├── TritonGPUTypeInterfaces.td │ │ │ ├── TritonGPUTypes.td │ │ │ └── Types.h │ │ └── Transforms │ │ │ ├── CMakeLists.txt │ │ │ ├── DecomposeScaledBlocked.h │ │ │ ├── MMAv5PipelineUtility.h │ │ │ ├── Partition.h │ │ │ ├── PartitionBuilder.h │ │ │ ├── Passes.h │ │ │ ├── Passes.td │ │ │ ├── PipelineExpander.h │ │ │ ├── PipeliningUtility.h │ │ │ ├── Schedule.h │ │ │ ├── TritonGPUConversion.h │ │ │ ├── Utility.h │ │ │ └── WarpSpecialization.h │ ├── TritonInstrument │ │ ├── CMakeLists.txt │ │ ├── IR │ │ │ ├── CMakeLists.txt │ │ │ ├── Dialect.h │ │ │ ├── TritonInstrumentAttrDefs.td │ │ │ ├── TritonInstrumentDialect.td │ │ │ ├── TritonInstrumentOps.td │ │ │ └── Utility.h │ │ └── Transforms │ │ │ ├── CMakeLists.txt │ │ │ ├── Passes.h │ │ │ └── Passes.td │ └── TritonNvidiaGPU │ │ ├── CMakeLists.txt │ │ ├── IR │ │ ├── CMakeLists.txt │ │ ├── Dialect.h │ │ ├── TritonNvidiaGPUAttrDefs.td │ │ ├── TritonNvidiaGPUDialect.td │ │ ├── TritonNvidiaGPUOpInterfaces.td │ │ └── TritonNvidiaGPUOps.td │ │ └── Transforms │ │ ├── CMakeLists.txt │ │ ├── Passes.h │ │ ├── Passes.td │ │ ├── TMAUtilities.h │ │ └── Utility.h │ ├── Target │ ├── CMakeLists.txt │ └── LLVMIR │ │ ├── CMakeLists.txt │ │ ├── Passes.h │ │ └── Passes.td │ └── Tools │ ├── GenericSwizzling.h │ ├── LayoutUtils.h │ ├── LinearLayout.h │ ├── StrUtil.h │ └── Sys │ └── GetEnv.hpp ├── lib ├── Analysis │ ├── Alias.cpp │ ├── Allocation.cpp │ ├── AxisInfo.cpp │ ├── CMakeLists.txt │ ├── Membar.cpp │ └── Utility.cpp ├── CMakeLists.txt ├── Conversion │ ├── CMakeLists.txt │ ├── TritonGPUToLLVM │ │ ├── AllocateSharedMemory.cpp │ │ ├── AllocateSharedMemoryUtility.cpp │ │ ├── AllocateWarpGroups.cpp │ │ ├── AssertOpToLLVM.cpp │ │ ├── CMakeLists.txt │ │ ├── ControlFlowOpToLLVM.cpp │ │ ├── ConvertLayoutOpToLLVM.cpp │ │ ├── DotOpToLLVM │ │ │ ├── FMA.cpp │ │ │ └── FMADotUtility.cpp │ │ ├── ElementwiseOpToLLVM.cpp │ │ ├── FuncOpToLLVM.cpp │ │ ├── GatherOpToLLVM.cpp │ │ ├── GlobalScratchMemoryAllocation.cpp │ │ ├── HistogramOpToLLVM.cpp │ │ ├── MakeRangeOpToLLVM.cpp │ │ ├── MemoryOpToLLVM.cpp │ │ ├── PrintOpToLLVM.cpp │ │ ├── ReduceOpToLLVM.cpp │ │ ├── ReduceScanCommon.h │ │ ├── SPMDOpToLLVM.cpp │ │ ├── ScanOpToLLVM.cpp │ │ ├── TypeConverter.cpp │ │ ├── Utility.cpp │ │ └── ViewOpToLLVM.cpp │ ├── TritonInstrumentToLLVM │ │ ├── CMakeLists.txt │ │ └── InstrumentationToLLVM.cpp │ └── TritonToTritonGPU │ │ ├── CMakeLists.txt │ │ ├── RelayoutTritonGPU.cpp │ │ ├── TritonGPUConversion.cpp │ │ └── TritonToTritonGPUPass.cpp ├── Dialect │ ├── CMakeLists.txt │ ├── Gluon │ │ ├── CMakeLists.txt │ │ ├── IR │ │ │ ├── CMakeLists.txt │ │ │ └── Dialect.cpp │ │ └── Transforms │ │ │ ├── CMakeLists.txt │ │ │ ├── Canonicalize.cpp │ │ │ ├── Inline.cpp │ │ │ └── ResolveAutoEncodings.cpp │ ├── Triton │ │ ├── CMakeLists.txt │ │ ├── IR │ │ │ ├── CMakeLists.txt │ │ │ ├── Canonicalize.td │ │ │ ├── Dialect.cpp │ │ │ ├── DiscardableAttributes.cpp │ │ │ ├── OpInterfaces.cpp │ │ │ ├── Ops.cpp │ │ │ ├── Traits.cpp │ │ │ ├── Types.cpp │ │ │ └── Utility.cpp │ │ └── Transforms │ │ │ ├── ArithTypeConversion.cpp │ │ │ ├── CMakeLists.txt │ │ │ ├── Combine.cpp │ │ │ ├── Combine.td │ │ │ ├── FunctionTypeConversion.cpp │ │ │ ├── LoopAwareCSE.cpp │ │ │ ├── LoopInvariantCodeMotion.cpp │ │ │ ├── LoopPeeling.cpp │ │ │ ├── LoopUnroll.cpp │ │ │ ├── ReorderBroadcast.cpp │ │ │ ├── RewriteTensorDescriptorToPointer.cpp │ │ │ └── RewriteTensorPointer.cpp │ ├── TritonGPU │ │ ├── CMakeLists.txt │ │ ├── IR │ │ │ ├── CMakeLists.txt │ │ │ ├── Dialect.cpp │ │ │ ├── LayoutUtility.cpp │ │ │ ├── LinearLayoutConversions.cpp │ │ │ ├── Ops.cpp │ │ │ └── Types.cpp │ │ └── Transforms │ │ │ ├── AccelerateMatmul.cpp │ │ │ ├── CMakeLists.txt │ │ │ ├── Coalesce.cpp │ │ │ ├── CoalesceAsyncCopy.cpp │ │ │ ├── CombineTensorSelectAndIf.cpp │ │ │ ├── DecomposeScaledBlocked.cpp │ │ │ ├── F32DotTC.cpp │ │ │ ├── FuseNestedLoops.cpp │ │ │ ├── HoistTMEMAlloc.cpp │ │ │ ├── OptimizeAccumulatorInit.cpp │ │ │ ├── OptimizeDotOperands.cpp │ │ │ ├── OptimizeThreadLocality.cpp │ │ │ ├── Pipeliner │ │ │ ├── AssignLatencies.cpp │ │ │ ├── LowerLoops.cpp │ │ │ ├── MMAv5PipelineUtility.cpp │ │ │ ├── PipelineExpander.cpp │ │ │ ├── PipeliningUtility.cpp │ │ │ ├── Schedule.cpp │ │ │ ├── ScheduleLoops.cpp │ │ │ ├── SoftwarePipeliner.cpp │ │ │ ├── TMAStoresPipeline.cpp │ │ │ ├── TestPipelineLowerLoop.cpp │ │ │ └── WGMMAPipeline.cpp │ │ │ ├── Prefetch.cpp │ │ │ ├── ReduceDataDuplication.cpp │ │ │ ├── RemoveLayoutConversions.cpp │ │ │ ├── ReorderInstructions.cpp │ │ │ ├── Utility.cpp │ │ │ └── WarpSpecialization │ │ │ ├── AutomaticWarpSpecialization.cpp │ │ │ ├── LoadMMASpecialization.cpp │ │ │ ├── OptimizePartitionWarps.cpp │ │ │ ├── Partition.cpp │ │ │ ├── PartitionBuilder.cpp │ │ │ ├── PartitionLoops.cpp │ │ │ ├── PartitionScheduling.cpp │ │ │ └── RewritePartitionDependencies.cpp │ ├── TritonInstrument │ │ ├── CMakeLists.txt │ │ ├── IR │ │ │ ├── CMakeLists.txt │ │ │ ├── Dialect.cpp │ │ │ ├── Ops.cpp │ │ │ └── Utility.cpp │ │ └── Transforms │ │ │ ├── CMakeLists.txt │ │ │ └── ConcurrencySanitizer.cpp │ └── TritonNvidiaGPU │ │ ├── CMakeLists.txt │ │ ├── IR │ │ ├── CMakeLists.txt │ │ ├── Dialect.cpp │ │ └── Ops.cpp │ │ └── Transforms │ │ ├── CMakeLists.txt │ │ ├── FenceInsertion.cpp │ │ ├── InterleaveTMem.cpp │ │ ├── MMALowering.cpp │ │ ├── OptimizeDescriptorEncoding.cpp │ │ ├── OptimizeTMemLayouts.cpp │ │ ├── PlanCTA.cpp │ │ ├── PromoteLHSToTMem.cpp │ │ ├── ProxFenceInsertion.cpp │ │ ├── RemoveTMEMTokens.cpp │ │ ├── TMALowering.cpp │ │ ├── TMAUtilities.cpp │ │ ├── TensorMemoryAllocation.cpp │ │ └── Utility.cpp ├── Instrumentation │ ├── CMakeLists.txt │ └── PrintLoadStoreMemSpaces.cpp ├── Target │ ├── CMakeLists.txt │ └── LLVMIR │ │ ├── CMakeLists.txt │ │ ├── LLVMDIScope.cpp │ │ ├── LLVMIRBreakPhiStruct.cpp │ │ └── LLVMPasses.h └── Tools │ ├── CMakeLists.txt │ ├── GenericSwizzling.cpp │ ├── LayoutUtils.cpp │ └── LinearLayout.cpp ├── pyproject.toml ├── python ├── build_helpers.py ├── examples │ └── gluon │ │ └── 01-attention-forward.py ├── requirements.txt ├── src │ ├── gluon_ir.cc │ ├── interpreter.cc │ ├── ir.cc │ ├── ir.h │ ├── llvm.cc │ ├── main.cc │ ├── passes.cc │ └── passes.h ├── test-requirements.txt ├── test │ ├── backend │ │ ├── extension_backend.c │ │ └── test_device_backend.py │ ├── conftest.py │ ├── gluon │ │ ├── test_consan.py │ │ ├── test_core.py │ │ ├── test_frontend.py │ │ └── test_lowerings.py │ ├── kernel_comparison │ │ └── kernels.yml │ ├── microbenchmark │ │ └── launch_overhead.py │ ├── regression │ │ ├── test_cast_matmul.py │ │ └── test_functional_regressions.py │ └── unit │ │ ├── cuda │ │ ├── __init__.py │ │ ├── test_experimental_tma.py │ │ ├── test_mixed_io.py │ │ ├── test_tensor_descriptor.py │ │ ├── test_tma_descriptor.py │ │ └── test_tma_store_gemm.py │ │ ├── instrumentation │ │ └── test_gpuhello.py │ │ ├── language │ │ ├── print_helper.py │ │ ├── test_annotations.py │ │ ├── test_block_pointer.py │ │ ├── test_compile_errors.py │ │ ├── test_compile_only.py │ │ ├── test_conversions.py │ │ ├── test_core.py │ │ ├── test_decorator.py │ │ ├── test_frontend.py │ │ ├── test_libdevice.py │ │ ├── test_line_info.py │ │ ├── test_matmul.py │ │ ├── test_module.py │ │ ├── test_mxfp.py │ │ ├── test_pipeliner.py │ │ ├── test_random.py │ │ ├── test_reproducer.py │ │ ├── test_standard.py │ │ ├── test_subprocess.py │ │ ├── test_tensor_descriptor.py │ │ ├── test_tlx.py │ │ ├── test_tuple.py │ │ └── test_warp_specialization.py │ │ ├── runtime │ │ ├── test_autotuner.py │ │ ├── test_bindings.py │ │ ├── test_build.py │ │ ├── test_cache.py │ │ ├── test_compilation_listener.py │ │ ├── test_cublas.py │ │ ├── test_driver.py │ │ ├── test_jit.py │ │ ├── test_launch.py │ │ └── test_subproc.py │ │ ├── test_debug.py │ │ ├── test_debug_dump.py │ │ ├── test_filecheck.py │ │ ├── test_knobs.py │ │ ├── test_link.py │ │ ├── test_perf_warning.py │ │ └── tools │ │ ├── test_aot.py │ │ ├── test_disasm.py │ │ └── test_irsource.py ├── triton │ ├── __init__.py │ ├── _filecheck.py │ ├── _internal_testing.py │ ├── _utils.py │ ├── backends │ │ ├── __init__.py │ │ ├── compiler.py │ │ └── driver.py │ ├── compiler │ │ ├── __init__.py │ │ ├── code_generator.py │ │ ├── compiler.py │ │ ├── errors.py │ │ └── make_launcher.py │ ├── errors.py │ ├── experimental │ │ ├── __init__.py │ │ └── gluon │ │ │ ├── __init__.py │ │ │ ├── _compiler.py │ │ │ ├── _runtime.py │ │ │ ├── language │ │ │ ├── __init__.py │ │ │ ├── _core.py │ │ │ ├── _layouts.py │ │ │ ├── _math.py │ │ │ ├── _semantic.py │ │ │ ├── _standard.py │ │ │ ├── amd │ │ │ │ ├── __init__.py │ │ │ │ ├── _layouts.py │ │ │ │ ├── cdna3 │ │ │ │ │ └── __init__.py │ │ │ │ └── cdna4 │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── async_copy.py │ │ │ ├── extra │ │ │ │ └── __init__.py │ │ │ └── nvidia │ │ │ │ ├── __init__.py │ │ │ │ ├── ampere │ │ │ │ ├── __init__.py │ │ │ │ ├── async_copy.py │ │ │ │ └── mbarrier.py │ │ │ │ ├── blackwell │ │ │ │ ├── __init__.py │ │ │ │ └── tma.py │ │ │ │ └── hopper │ │ │ │ ├── __init__.py │ │ │ │ ├── mbarrier.py │ │ │ │ └── tma.py │ │ │ └── nvidia │ │ │ ├── __init__.py │ │ │ ├── blackwell.py │ │ │ └── hopper.py │ ├── knobs.py │ ├── language │ │ ├── __init__.py │ │ ├── core.py │ │ ├── extra │ │ │ ├── __init__.py │ │ │ └── libdevice.py │ │ ├── math.py │ │ ├── random.py │ │ ├── semantic.py │ │ ├── standard.py │ │ └── target_info.py │ ├── runtime │ │ ├── __init__.py │ │ ├── _allocation.py │ │ ├── _async_compile.py │ │ ├── autotuner.py │ │ ├── build.py │ │ ├── cache.py │ │ ├── driver.py │ │ ├── errors.py │ │ ├── interpreter.py │ │ └── jit.py │ ├── testing.py │ └── tools │ │ ├── __init__.py │ │ ├── build_extern.py │ │ ├── compile.py │ │ ├── disasm.py │ │ ├── experimental_descriptor.py │ │ ├── link.py │ │ ├── mxfp.py │ │ ├── ragged_tma.py │ │ └── tensor_descriptor.py ├── triton_kernels │ ├── .gitignore │ ├── bench │ │ ├── bench_mlp.py │ │ ├── bench_utils.py │ │ └── distributed.py │ ├── pyproject.toml │ ├── tests │ │ ├── __init__.py │ │ ├── conftest.py │ │ ├── test_compaction.py │ │ ├── test_matmul.py │ │ ├── test_mxfp.py │ │ ├── test_routing.py │ │ ├── test_specialize.py │ │ ├── test_swiglu.py │ │ ├── test_tensor.py │ │ └── test_tensor_details │ │ │ ├── test_layout_blackwell.py │ │ │ └── test_layout_hopper.py │ └── triton_kernels │ │ ├── __init__.py │ │ ├── compaction.py │ │ ├── compaction_details │ │ └── _masked_compaction.py │ │ ├── matmul_ogs.py │ │ ├── matmul_ogs_details │ │ ├── _common.py │ │ ├── _matmul_ogs.py │ │ ├── _p_matmul_ogs.py │ │ ├── _reduce_grouped.py │ │ ├── opt_flags.py │ │ └── opt_flags_details │ │ │ ├── opt_flags_amd.py │ │ │ └── opt_flags_nvidia.py │ │ ├── numerics.py │ │ ├── numerics_details │ │ ├── __init__.py │ │ ├── flexpoint.py │ │ ├── mxfp.py │ │ └── mxfp_details │ │ │ ├── _downcast_to_mxfp.py │ │ │ └── _upcast_from_mxfp.py │ │ ├── proton_opts.py │ │ ├── reduction_details │ │ └── reduce_bitmatrix.py │ │ ├── roofline.py │ │ ├── routing.py │ │ ├── routing_details │ │ ├── _expt_data.py │ │ └── _routing_compute.py │ │ ├── specialize.py │ │ ├── swiglu.py │ │ ├── swiglu_details │ │ └── _swiglu.py │ │ ├── target_info.py │ │ ├── tensor.py │ │ ├── tensor_details │ │ ├── layout.py │ │ └── layout_details │ │ │ ├── base.py │ │ │ ├── blackwell_scale.py │ │ │ ├── blackwell_value.py │ │ │ ├── cdna4_scale.py │ │ │ ├── hopper_scale.py │ │ │ ├── hopper_value.py │ │ │ └── strided.py │ │ ├── testing.py │ │ ├── topk.py │ │ └── topk_details │ │ ├── __init__.py │ │ ├── _topk_backward.py │ │ └── _topk_forward.py └── tutorials │ ├── 01-vector-add.py │ ├── 02-fused-softmax.py │ ├── 03-matrix-multiplication.py │ ├── 04-low-memory-dropout.py │ ├── 05-layer-norm.py │ ├── 06-fused-attention.py │ ├── 07-extern-functions.py │ ├── 08-grouped-gemm.py │ ├── 09-persistent-matmul.py │ ├── 10-block-scaled-matmul.py │ ├── 11-programmatic-dependent-launch.py │ ├── README.rst │ ├── fused-attention-ws-device-tma.py │ ├── fused-attention-ws.py │ └── gluon │ ├── 01-intro.py │ ├── 02-layouts.py │ ├── 03-async-copy.py │ ├── 04-tma.py │ ├── 05-wgmma.py │ ├── 06-tcgen05.py │ ├── 07-persistence.py │ └── 08-warp-specialization.py ├── scripts └── build-llvm-project.sh ├── setup.py ├── test ├── Analysis │ ├── amd │ │ └── test-alignment.mlir │ ├── test-alias.mlir │ ├── test-alignment.mlir │ ├── test-allocation.mlir │ ├── test-membar-ttng.mlir │ ├── test-membar.mlir │ └── test-transpose-axisinfo.mlir ├── CMakeLists.txt ├── Conversion │ ├── allocate_shared_memory.mlir │ ├── allocate_warp_groups.mlir │ ├── amd │ │ ├── allocate_shared_memory.mlir │ │ ├── amdgpu_membar.mlir │ │ ├── async-ops-alias-scopes.mlir │ │ ├── async_ops_to_llvm.mlir │ │ ├── async_ops_to_llvm_invalid.mlir │ │ ├── buffer_atomic_cas.mlir │ │ ├── buffer_load_store.mlir │ │ ├── buffer_load_to_local_to_llvm.mlir │ │ ├── builtin_func_to_llvm.mlir │ │ ├── compute-base-ptr.mlir │ │ ├── convert_layout.mlir │ │ ├── dedup-by-constancy.mlir │ │ ├── ds_transpose.mlir │ │ ├── fp_to_fp.mlir │ │ ├── in_thread_transpose.mlir │ │ ├── invalid_concat_op.mlir │ │ ├── invalid_extractslice_to_llvm.mlir │ │ ├── load_store.mlir │ │ ├── math-denorm-handling.mlir │ │ ├── mfma-shortcut.mlir │ │ ├── minmax.mlir │ │ ├── tritongpu_to_llvm.mlir │ │ ├── tritongpu_to_llvm_rdna.mlir │ │ ├── tritongpu_wmma_dot_to_llvm.mlir │ │ ├── upcast_mxfp.mlir │ │ ├── wmma-v1-shortcut.mlir │ │ └── wmma-v2-shortcut.mlir │ ├── atomic_ldst.mlir │ ├── cvt_to_llvm.mlir │ ├── dedup-by-constancy.mlir │ ├── divide-by-0.mlir │ ├── nvgpu_to_llvm.mlir │ ├── reduce_to_llvm.mlir │ ├── relayout_tritongpu.mlir │ ├── scan_to_llvm.mlir │ ├── tma_to_llvm.mlir │ ├── triton_to_tritongpu.mlir │ ├── tritongpu_to_llvm.mlir │ ├── tritongpu_to_llvm_blackwell.mlir │ ├── tritongpu_to_llvm_block_dot_shortcut.mlir │ ├── tritongpu_to_llvm_debug.mlir │ ├── tritongpu_to_llvm_hopper.mlir │ ├── tritongpu_to_llvm_hopper_ptx80.mlir │ ├── tritongpu_to_llvm_sm120.mlir │ ├── tritongpu_to_llvm_volta.mlir │ ├── tritongpu_to_ptx.mlir │ ├── tritongpu_to_ptx_mmav3.mlir │ ├── tritoninstrument_to_llvm.mlir │ ├── tritonnvidiagpu_to_llvm.mlir │ ├── ttg_warp_specialize.mlir │ └── warp_specialize_to_llvm.mlir ├── Gluon │ ├── auto_encoding.mlir │ ├── inlining.mlir │ └── invalid_auto_encoding.mlir ├── Hopper │ ├── CMakeLists.txt │ └── WarpSpecialization │ │ ├── 1D_tmem.mlir │ │ ├── blackwell_ws_data_partition.mlir │ │ ├── fa_code_partition.mlir │ │ ├── ws_code_partition.mlir │ │ ├── ws_data_partition.mlir │ │ ├── ws_task_id_propagation.mlir │ │ └── ws_task_partition.mlir ├── LLVMIR │ └── break-phi-struct.ll ├── NVWS │ ├── assign_stage_phase.mlir │ ├── insert_aref.mlir │ ├── invalid.mlir │ ├── lower_aref.mlir │ ├── lower_warp_group.mlir │ └── ops.mlir ├── Proton │ ├── allocate_global_scratch_buffer.mlir │ ├── allocate_shared_memory.mlir │ ├── amd │ │ ├── add_sched_barriers.mlir │ │ └── protongpu_to_llvm.mlir │ ├── nvidia │ │ └── protongpu_to_llvm.mlir │ ├── ops.mlir │ ├── proton_to_protongpu.mlir │ ├── protongpu_transforms.mlir │ └── scope_id.mlir ├── TLX │ ├── attach-metadata.mlir │ ├── insert-require-layout.mlir │ ├── insert_cluster_sync_ops.mlir │ ├── ops.mlir │ ├── propagate-layout.mlir │ ├── rewrite-local-alias.mlir │ └── tlx-verifier.mlir ├── Tools │ └── tensor_layout_print.mlir ├── Triton │ ├── canonicalize.mlir │ ├── combine.mlir │ ├── invalid.mlir │ ├── loop-invariant-code-motion.mlir │ ├── loop-peeling.mlir │ ├── loop-unroll.mlir │ ├── loop_cse.mlir │ ├── ops.mlir │ ├── reorder-broadcast.mlir │ ├── reproducer.mlir │ ├── rewrite-tensor-descriptor-to-pointer.mlir │ ├── rewrite-tensor-pointer.mlir │ ├── vecadd.mlir │ └── verify-make-range.mlir ├── TritonGPU │ ├── accelerate-matmul.mlir │ ├── accumulator-init.mlir │ ├── amd │ │ ├── accelerate-amd-matmul-chain-dot.mlir │ │ ├── accelerate-amd-matmul-fma.mlir │ │ ├── accelerate-amd-matmul-mfma-gfx950.mlir │ │ ├── accelerate-amd-matmul-mfma.mlir │ │ ├── accelerate-amd-matmul-wmma-gen1.mlir │ │ ├── accelerate-amd-matmul-wmma-gen2.mlir │ │ ├── amd-block-pingpong-chained-dots.mlir │ │ ├── amd-block-pingpong.mlir │ │ ├── amd-canonicalize-extract-slice.mlir │ │ ├── amd-canonicalize-pointers-dont-run-mlir-canonicalizer.mlir │ │ ├── amd-canonicalize-pointers-no-large-tensor.mlir │ │ ├── amd-canonicalize-pointers.mlir │ │ ├── amd-coalesce-async-copy.mlir │ │ ├── amd-concat-op.mlir │ │ ├── amd-conditional-barrier.mlir │ │ ├── amd-convert-buffer-ops-range-analysis.mlir │ │ ├── amd-convert-buffer-ops-small-tensor.mlir │ │ ├── amd-convert-buffer-ops.mlir │ │ ├── amd-extractslice-op.mlir │ │ ├── amd-fold-true-cmpi.mlir │ │ ├── amd-hoist-cvtToDotOp.mlir │ │ ├── amd-optimize-dot-operands.mlir │ │ ├── amd-optimize-epilogue.mlir │ │ ├── amd-pipeline-chained-dots.mlir │ │ ├── amd-range-analysis.mlir │ │ ├── amd-reorder-instructions.mlir │ │ ├── amd-schedule-hint.mlir │ │ ├── amd-stream-loop-assume.mlir │ │ ├── amd-stream-prefetch.mlir │ │ ├── amd-update-async-wait-count.mlir │ │ ├── in-thread-transpose.mlir │ │ ├── invalid.mlir │ │ ├── mfma-double-rate.mlir │ │ ├── mfma-xf32.mlir │ │ ├── optimize-lds-usage.mlir │ │ └── sink-setprio-mfma.mlir │ ├── atomic-cas.mlir │ ├── attention-dp-loop-schedule.mlir │ ├── automatic-warp-specialization.mlir │ ├── bf16x3-matmul.mlir │ ├── canonicalize.mlir │ ├── coalesce-async-copy.mlir │ ├── coalesce.mlir │ ├── combine-select-if.mlir │ ├── combine.mlir │ ├── consan-negative.mlir │ ├── consan.mlir │ ├── dot-operands.mlir │ ├── fence-inserstion.mlir │ ├── fuse-nested-loops.mlir │ ├── global_scratch_alloc.mlir │ ├── global_scratch_to_llvm.mlir │ ├── hoist-tmem-alloc.mlir │ ├── inline.mlir │ ├── invalid-attributes.mlir │ ├── invalid.mlir │ ├── load-mma-specialization.mlir │ ├── loop-pipeline-async-latencies.mlir │ ├── loop-pipeline-blackwell.mlir │ ├── loop-pipeline-combine-waits.mlir │ ├── loop-pipeline-cuda.mlir │ ├── loop-pipeline-expand.mlir │ ├── loop-pipeline-hip.mlir │ ├── loop-pipeline-hopper-remove-wait.mlir │ ├── loop-pipeline-hopper.mlir │ ├── loop-pipeline-indirect-load.mlir │ ├── loop-pipeline.mlir │ ├── loop-schedule.mlir │ ├── matmul-loop-pipeline.mlir │ ├── matmul.mlir │ ├── memdesc-subview-split.mlir │ ├── metaws-loop-schedule.mlir │ ├── ops.mlir │ ├── optimize-locality.mlir │ ├── optimize-partition-warps.mlir │ ├── optimize_epilogue.mlir │ ├── partition-loops.mlir │ ├── partition-scheduling.mlir │ ├── pipeline-assign-latencies.mlir │ ├── pipeline-loop-nest.mlir │ ├── pipeline-lower-loop.mlir │ ├── pipeline-schedule-loop.mlir │ ├── prefetch.mlir │ ├── promote-lhs-to-tmem.mlir │ ├── prox_fence_insertion.mlir │ ├── reduce-data-duplication.mlir │ ├── reorder-instructions.mlir │ ├── rewrite-partition-dependencies.mlir │ ├── samples │ │ ├── descriptor-matmul-pipeline.mlir │ │ ├── descriptor-matmul-pipeline.mlir.in │ │ ├── simulated-grouped-gemm.mlir │ │ └── simulated-grouped-gemm.mlir.in │ ├── tf32x3-matmul.mlir │ └── verify-blocked-layout.mlir ├── TritonNvidiaGPU │ ├── bf16-atomics.mlir │ ├── canonicalize.mlir │ ├── inline.mlir │ ├── interleave_tmem.mlir │ ├── invalid.mlir │ ├── membar.mlir │ ├── mma_lowering.mlir │ ├── ops.mlir │ ├── optimize_descriptor_encoding.mlir │ ├── test_promotion_to_tensor_memory.mlir │ ├── test_tensor_memory_allocation.mlir │ ├── tma_lowering.mlir │ └── tmem_layouts.mlir ├── include │ └── Analysis │ │ └── TestAxisInfo.h ├── lib │ ├── Analysis │ │ ├── CMakeLists.txt │ │ ├── TestAlias.cpp │ │ ├── TestAllocation.cpp │ │ ├── TestAxisInfo.cpp │ │ ├── TestMembar.cpp │ │ └── TestPrintNesting.cpp │ ├── CMakeLists.txt │ ├── Dialect │ │ ├── CMakeLists.txt │ │ └── TestLoopPeeling.cpp │ ├── Instrumentation │ │ ├── CMakeLists.txt │ │ └── GPUHello.cpp │ └── Proton │ │ ├── CMakeLists.txt │ │ └── TestScopeIdAllocation.cpp ├── lit.cfg.py └── lit.site.cfg.py.in ├── third_party ├── amd │ ├── CMakeLists.txt │ ├── backend │ │ ├── __init__.py │ │ ├── compiler.py │ │ ├── driver.c │ │ ├── driver.py │ │ ├── include │ │ │ ├── hip │ │ │ │ ├── amd_detail │ │ │ │ │ ├── amd_channel_descriptor.h │ │ │ │ │ ├── amd_device_functions.h │ │ │ │ │ ├── amd_hip_atomic.h │ │ │ │ │ ├── amd_hip_bf16.h │ │ │ │ │ ├── amd_hip_bfloat16.h │ │ │ │ │ ├── amd_hip_common.h │ │ │ │ │ ├── amd_hip_complex.h │ │ │ │ │ ├── amd_hip_cooperative_groups.h │ │ │ │ │ ├── amd_hip_fp16.h │ │ │ │ │ ├── amd_hip_fp8.h │ │ │ │ │ ├── amd_hip_gl_interop.h │ │ │ │ │ ├── amd_hip_math_constants.h │ │ │ │ │ ├── amd_hip_runtime.h │ │ │ │ │ ├── amd_hip_runtime_pt_api.h │ │ │ │ │ ├── amd_hip_unsafe_atomics.h │ │ │ │ │ ├── amd_hip_vector_types.h │ │ │ │ │ ├── amd_math_functions.h │ │ │ │ │ ├── amd_surface_functions.h │ │ │ │ │ ├── amd_warp_functions.h │ │ │ │ │ ├── amd_warp_sync_functions.h │ │ │ │ │ ├── concepts.hpp │ │ │ │ │ ├── device_library_decls.h │ │ │ │ │ ├── functional_grid_launch.hpp │ │ │ │ │ ├── grid_launch.h │ │ │ │ │ ├── grid_launch.hpp │ │ │ │ │ ├── grid_launch_GGL.hpp │ │ │ │ │ ├── helpers.hpp │ │ │ │ │ ├── hip_api_trace.hpp │ │ │ │ │ ├── hip_assert.h │ │ │ │ │ ├── hip_cooperative_groups_helper.h │ │ │ │ │ ├── hip_fp16_gcc.h │ │ │ │ │ ├── hip_fp16_math_fwd.h │ │ │ │ │ ├── hip_ldg.h │ │ │ │ │ ├── hip_prof_str.h │ │ │ │ │ ├── hip_runtime_prof.h │ │ │ │ │ ├── host_defines.h │ │ │ │ │ ├── hsa_helpers.hpp │ │ │ │ │ ├── macro_based_grid_launch.hpp │ │ │ │ │ ├── math_fwd.h │ │ │ │ │ ├── ockl_image.h │ │ │ │ │ ├── program_state.hpp │ │ │ │ │ ├── texture_fetch_functions.h │ │ │ │ │ └── texture_indirect_functions.h │ │ │ │ ├── channel_descriptor.h │ │ │ │ ├── device_functions.h │ │ │ │ ├── driver_types.h │ │ │ │ ├── hip_bf16.h │ │ │ │ ├── hip_bfloat16.h │ │ │ │ ├── hip_common.h │ │ │ │ ├── hip_complex.h │ │ │ │ ├── hip_cooperative_groups.h │ │ │ │ ├── hip_deprecated.h │ │ │ │ ├── hip_ext.h │ │ │ │ ├── hip_fp16.h │ │ │ │ ├── hip_fp8.h │ │ │ │ ├── hip_gl_interop.h │ │ │ │ ├── hip_hcc.h │ │ │ │ ├── hip_math_constants.h │ │ │ │ ├── hip_profile.h │ │ │ │ ├── hip_runtime.h │ │ │ │ ├── hip_runtime_api.h │ │ │ │ ├── hip_texture_types.h │ │ │ │ ├── hip_vector_types.h │ │ │ │ ├── hip_version.h │ │ │ │ ├── hiprtc.h │ │ │ │ ├── library_types.h │ │ │ │ ├── math_functions.h │ │ │ │ ├── surface_types.h │ │ │ │ └── texture_types.h │ │ │ ├── hsa │ │ │ │ ├── Brig.h │ │ │ │ ├── amd_hsa_common.h │ │ │ │ ├── amd_hsa_elf.h │ │ │ │ ├── amd_hsa_kernel_code.h │ │ │ │ ├── amd_hsa_queue.h │ │ │ │ ├── amd_hsa_signal.h │ │ │ │ ├── hsa.h │ │ │ │ ├── hsa_amd_tool.h │ │ │ │ ├── hsa_api_trace.h │ │ │ │ ├── hsa_api_trace_version.h │ │ │ │ ├── hsa_ext_amd.h │ │ │ │ ├── hsa_ext_finalize.h │ │ │ │ ├── hsa_ext_image.h │ │ │ │ ├── hsa_ven_amd_aqlprofile.h │ │ │ │ ├── hsa_ven_amd_loader.h │ │ │ │ └── hsa_ven_amd_pc_sampling.h │ │ │ └── roctracer │ │ │ │ ├── ext │ │ │ │ └── prof_protocol.h │ │ │ │ ├── hip_ostream_ops.h │ │ │ │ ├── hsa_ostream_ops.h │ │ │ │ ├── hsa_prof_str.h │ │ │ │ ├── roctracer.h │ │ │ │ ├── roctracer_ext.h │ │ │ │ ├── roctracer_hcc.h │ │ │ │ ├── roctracer_hip.h │ │ │ │ ├── roctracer_hsa.h │ │ │ │ ├── roctracer_plugin.h │ │ │ │ ├── roctracer_roctx.h │ │ │ │ └── roctx.h │ │ └── lib │ │ │ ├── asanrtl.bc │ │ │ ├── ockl.bc │ │ │ └── ocml.bc │ ├── include │ │ ├── Analysis │ │ │ ├── AMDGPUAllocation.h │ │ │ ├── AxisInfoExt.h │ │ │ └── RangeAnalysis.h │ │ ├── CMakeLists.txt │ │ ├── Dialect │ │ │ ├── CMakeLists.txt │ │ │ └── TritonAMDGPU │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── IR │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── Dialect.h │ │ │ │ ├── TritonAMDGPUAttrDefs.td │ │ │ │ ├── TritonAMDGPUDialect.td │ │ │ │ └── TritonAMDGPUOps.td │ │ │ │ └── Utility │ │ │ │ └── CommonUtils.h │ │ ├── TritonAMDGPUToLLVM │ │ │ ├── CMakeLists.txt │ │ │ ├── GCNAsmFormat.h │ │ │ ├── MembarUtility.h │ │ │ ├── Passes.h │ │ │ ├── Passes.td │ │ │ ├── PatternTritonAMDGPUToLLVM.h │ │ │ └── TargetUtils.h │ │ ├── TritonAMDGPUTransforms │ │ │ ├── CMakeLists.txt │ │ │ ├── MfmaGroup.h │ │ │ ├── Passes.h │ │ │ ├── Passes.td │ │ │ ├── TritonGPUConversion.h │ │ │ └── WmmaGroup.h │ │ └── Utils │ │ │ └── Utility.h │ ├── language │ │ └── hip │ │ │ ├── __init__.py │ │ │ ├── libdevice.py │ │ │ └── utils.py │ ├── lib │ │ ├── Analysis │ │ │ ├── AMDGPUAllocation.cpp │ │ │ ├── AxisInfoExt.cpp │ │ │ ├── CMakeLists.txt │ │ │ └── RangeAnalysis.cpp │ │ ├── CMakeLists.txt │ │ ├── Dialect │ │ │ ├── CMakeLists.txt │ │ │ └── TritonAMDGPU │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── IR │ │ │ │ ├── CMakeLists.txt │ │ │ │ └── Dialect.cpp │ │ │ │ └── Utility │ │ │ │ ├── CMakeLists.txt │ │ │ │ └── CommonUtils.cpp │ │ ├── TritonAMDGPUDialectToLLVM │ │ │ ├── CMakeLists.txt │ │ │ ├── ConcatOpToLLVM.cpp │ │ │ ├── ExtractSliceOpToLLVM.cpp │ │ │ ├── InThreadTransposeOpToTTG.cpp │ │ │ ├── TritonAMDGPUToLLVMPatterns.cpp │ │ │ ├── Utility.cpp │ │ │ └── Utility.h │ │ ├── TritonAMDGPUToLLVM │ │ │ ├── AllocateSharedMemory.cpp │ │ │ ├── AsyncUtility.cpp │ │ │ ├── AsyncUtility.h │ │ │ ├── AtomicRMWOpsEmitter.cpp │ │ │ ├── AtomicRMWOpsEmitter.h │ │ │ ├── BarrierOpConversion.cpp │ │ │ ├── BufferOpsEmitter.cpp │ │ │ ├── BufferOpsEmitter.h │ │ │ ├── BuiltinFuncToLLVM.cpp │ │ │ ├── CMakeLists.txt │ │ │ ├── ConvertLayoutOpToLLVM.cpp │ │ │ ├── DotOpToLLVM.cpp │ │ │ ├── DotOpToLLVM │ │ │ │ ├── FMA.cpp │ │ │ │ ├── MFMA.cpp │ │ │ │ └── WMMA.cpp │ │ │ ├── ElementwiseOpToLLVM.cpp │ │ │ ├── Fp4ToFpOpToLLVM.cpp │ │ │ ├── GCNAsmFormat.cpp │ │ │ ├── LoadStoreOpToLLVM.cpp │ │ │ ├── MembarUtility.cpp │ │ │ ├── MemoryOpToLLVM.cpp │ │ │ ├── OptimizeLDSUsage.cpp │ │ │ ├── OptimizeLDSUtility.cpp │ │ │ ├── OptimizeLDSUtility.h │ │ │ ├── PatternTritonGPUOpToLLVM.h │ │ │ ├── SPMDOpToLLVM.cpp │ │ │ ├── ScalarizePackedFOps.cpp │ │ │ ├── SchedInstructions.cpp │ │ │ ├── TargetInfo.cpp │ │ │ ├── TargetInfo.h │ │ │ ├── TargetUtils.cpp │ │ │ ├── TritonGPUToLLVM.cpp │ │ │ ├── UpcastMXFPToLLVM.cpp │ │ │ ├── Utility.cpp │ │ │ └── Utility.h │ │ └── TritonAMDGPUTransforms │ │ │ ├── AccelerateAMDMatmul.cpp │ │ │ ├── BlockPingpong.cpp │ │ │ ├── CMakeLists.txt │ │ │ ├── CanonicalizePointers.cpp │ │ │ ├── CoalesceAsyncCopy.cpp │ │ │ ├── ConvertToBufferOps.cpp │ │ │ ├── FoldTrueCmpIOp.cpp │ │ │ ├── HoistLayoutConversions.cpp │ │ │ ├── InThreadTranspose.cpp │ │ │ ├── LowerBarrierOps.cpp │ │ │ ├── MfmaGroup.cpp │ │ │ ├── OptimizeDotOperands.cpp │ │ │ ├── OptimizeEpilogue.cpp │ │ │ ├── ReorderInstructions.cpp │ │ │ ├── StreamPipeline.cpp │ │ │ ├── UpdateAsyncWaitCount.cpp │ │ │ ├── Utility.cpp │ │ │ ├── Utility.h │ │ │ └── WmmaGroup.cpp │ ├── python │ │ ├── test │ │ │ ├── address_sanitizer_helper.py │ │ │ ├── attn_fwd.ttir │ │ │ ├── conftest.py │ │ │ ├── test_address_sanitizer.py │ │ │ ├── test_convert_op_permlane_swap.py │ │ │ ├── test_extract_slice_concat_op.py │ │ │ └── test_scalarize_packed_fops.py │ │ └── triton_amd.cc │ ├── test │ │ ├── CMakeLists.txt │ │ └── lib │ │ │ ├── Analysis │ │ │ ├── CMakeLists.txt │ │ │ ├── TestAMDGPUMembar.cpp │ │ │ ├── TestAMDRangeAnalysis.cpp │ │ │ └── TestAxisInfo.cpp │ │ │ └── CMakeLists.txt │ ├── tools │ │ └── hip │ │ │ ├── compile.cpp │ │ │ └── compile.h │ └── unittest │ │ ├── CMakeLists.txt │ │ └── Conversion │ │ ├── CMakeLists.txt │ │ └── OptimizeLDSTest.cpp ├── f2reduce │ ├── CMakeLists.txt │ ├── LICENCE.txt │ ├── README.md │ ├── VERSION │ ├── f2reduce.cpp │ └── f2reduce.h ├── nvidia │ ├── CMakeLists.txt │ ├── backend │ │ ├── __init__.py │ │ ├── bin │ │ │ └── ptxas-blackwell │ │ ├── compiler.py │ │ ├── driver.c │ │ ├── driver.py │ │ └── lib │ │ │ └── libdevice.10.bc │ ├── hopper │ │ ├── CMakeLists.txt │ │ ├── include │ │ │ ├── CMakeLists.txt │ │ │ └── Transforms │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── Passes.h │ │ │ │ └── Passes.td │ │ └── lib │ │ │ ├── CMakeLists.txt │ │ │ └── Transforms │ │ │ ├── CMakeLists.txt │ │ │ ├── WarpSpecialization.cpp │ │ │ └── WarpSpecialization │ │ │ ├── CodePartitionUtility.cpp │ │ │ ├── CodePartitionUtility.h │ │ │ ├── PingPong.cpp │ │ │ ├── TMEMAlloc1D.cpp │ │ │ ├── TMEMUtils.h │ │ │ ├── TaskIdPropagation.cpp │ │ │ ├── TaskIdPropagation.h │ │ │ ├── Utility.cpp │ │ │ ├── Utility.h │ │ │ ├── WSBuffer.cpp │ │ │ ├── WSCodePartition.cpp │ │ │ ├── WSDataPartition.cpp │ │ │ ├── WSLowerMem.cpp │ │ │ ├── WSLowerToken.cpp │ │ │ ├── WSMemoryPlanner.cpp │ │ │ ├── WSSpecialize.cpp │ │ │ ├── WSTaskIdPropagate.cpp │ │ │ └── WSTaskPartition.cpp │ ├── include │ │ ├── CMakeLists.txt │ │ ├── Dialect │ │ │ ├── CMakeLists.txt │ │ │ ├── NVGPU │ │ │ │ ├── CMakeLists.txt │ │ │ │ └── IR │ │ │ │ │ ├── CMakeLists.txt │ │ │ │ │ ├── Dialect.h │ │ │ │ │ ├── NVGPUAttrDefs.td │ │ │ │ │ ├── NVGPUDialect.td │ │ │ │ │ └── NVGPUOps.td │ │ │ └── NVWS │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── IR │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── Dialect.h │ │ │ │ ├── NVWSAttrDefs.td │ │ │ │ ├── NVWSDialect.td │ │ │ │ ├── NVWSOpInterfaces.td │ │ │ │ ├── NVWSOps.td │ │ │ │ └── NVWSTypes.td │ │ │ │ └── Transforms │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── Passes.h │ │ │ │ └── Passes.td │ │ ├── NVGPUToLLVM │ │ │ ├── CMakeLists.txt │ │ │ ├── NVGPUToLLVMPass.h │ │ │ ├── Passes.h │ │ │ └── Passes.td │ │ ├── TritonNVIDIAGPUToLLVM │ │ │ ├── CMakeLists.txt │ │ │ ├── PTXAsmFormat.h │ │ │ ├── Passes.h │ │ │ ├── Passes.td │ │ │ └── Utility.h │ │ ├── cublas_instance.h │ │ └── cublas_types.h │ ├── language │ │ └── cuda │ │ │ ├── __init__.py │ │ │ ├── _experimental_tma.py │ │ │ ├── gdc.py │ │ │ ├── libdevice.py │ │ │ └── utils.py │ ├── lib │ │ ├── CMakeLists.txt │ │ ├── Dialect │ │ │ ├── CMakeLists.txt │ │ │ ├── NVGPU │ │ │ │ ├── CMakeLists.txt │ │ │ │ └── IR │ │ │ │ │ ├── CMakeLists.txt │ │ │ │ │ └── Dialect.cpp │ │ │ └── NVWS │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── IR │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── Dialect.cpp │ │ │ │ └── Ops.cpp │ │ │ │ └── Transforms │ │ │ │ ├── AssignStagePhase.cpp │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── InsertAref.cpp │ │ │ │ ├── LowerAref.cpp │ │ │ │ ├── LowerWarpGroup.cpp │ │ │ │ ├── Utilities.cpp │ │ │ │ └── Utilities.h │ │ ├── NVGPUToLLVM │ │ │ ├── CMakeLists.txt │ │ │ └── NVGPUToLLVMPass.cpp │ │ └── TritonNVIDIAGPUToLLVM │ │ │ ├── Allocation.cpp │ │ │ ├── Allocation.h │ │ │ ├── BarrierOpToLLVM.cpp │ │ │ ├── CMakeLists.txt │ │ │ ├── ClusterOpsToLLVM.cpp │ │ │ ├── ConvertLayoutOpToLLVM.cpp │ │ │ ├── ConvertWarpSpecializeToLLVM.cpp │ │ │ ├── DotOpToLLVM.cpp │ │ │ ├── DotOpToLLVM │ │ │ ├── MMAHelpers.h │ │ │ ├── MMAv2.cpp │ │ │ ├── MMAv5.cpp │ │ │ └── WGMMA.cpp │ │ │ ├── ElementwiseOpToLLVM.cpp │ │ │ ├── Fp4ToFpOpToLLVM.cpp │ │ │ ├── LoadStoreOpToLLVM.cpp │ │ │ ├── MemoryOpToLLVM.cpp │ │ │ ├── PTXAsmFormat.cpp │ │ │ ├── PatternTritonGPUOpToLLVM.h │ │ │ ├── SPMDOpToLLVM.cpp │ │ │ ├── TMAToLLVM.cpp │ │ │ ├── TargetInfo.cpp │ │ │ ├── TargetInfo.h │ │ │ ├── TensorMemoryToLLVM.cpp │ │ │ ├── TensorPtrOpsToLLVM.cpp │ │ │ ├── TritonGPUToLLVM.cpp │ │ │ ├── Utility.cpp │ │ │ └── Utility.h │ ├── tools │ │ └── cuda │ │ │ ├── compile.c │ │ │ └── compile.h │ ├── triton_nvidia.cc │ └── unittest │ │ ├── CMakeLists.txt │ │ └── Conversion │ │ ├── CMakeLists.txt │ │ └── TritonGPUToLLVM │ │ ├── CMakeLists.txt │ │ └── PTXAsmFormatTest.cpp ├── proton │ ├── .gitignore │ ├── CMakeLists.txt │ ├── Dialect │ │ ├── CMakeLists.txt │ │ ├── include │ │ │ ├── Analysis │ │ │ │ └── ScopeIdAllocation.h │ │ │ ├── CMakeLists.txt │ │ │ ├── Conversion │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── ProtonGPUToLLVM │ │ │ │ │ ├── CMakeLists.txt │ │ │ │ │ ├── Passes.h │ │ │ │ │ ├── Passes.td │ │ │ │ │ ├── PatternProtonGPUOpToLLVM.h │ │ │ │ │ ├── ProtonAMDGPUToLLVM │ │ │ │ │ │ ├── AMDPatternProtonGPUOpToLLVM.h │ │ │ │ │ │ ├── CMakeLists.txt │ │ │ │ │ │ ├── Passes.h │ │ │ │ │ │ ├── Passes.td │ │ │ │ │ │ └── TargetInfo.h │ │ │ │ │ ├── ProtonNvidiaGPUToLLVM │ │ │ │ │ │ ├── CMakeLists.txt │ │ │ │ │ │ ├── NvidiaPatternProtonGPUOpToLLVM.h │ │ │ │ │ │ ├── Passes.h │ │ │ │ │ │ ├── Passes.td │ │ │ │ │ │ └── TargetInfo.h │ │ │ │ │ ├── TargetInfoBase.h │ │ │ │ │ └── Utility.h │ │ │ │ └── ProtonToProtonGPU │ │ │ │ │ ├── CMakeLists.txt │ │ │ │ │ ├── Passes.h │ │ │ │ │ └── Passes.td │ │ │ └── Dialect │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── Proton │ │ │ │ ├── CMakeLists.txt │ │ │ │ └── IR │ │ │ │ │ ├── CMakeLists.txt │ │ │ │ │ ├── Dialect.h │ │ │ │ │ ├── ProtonAttrDefs.td │ │ │ │ │ ├── ProtonDialect.td │ │ │ │ │ └── ProtonOps.td │ │ │ │ └── ProtonGPU │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── IR │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── Dialect.h │ │ │ │ ├── ProtonGPUAttrDefs.td │ │ │ │ ├── ProtonGPUDialect.td │ │ │ │ ├── ProtonGPUOps.td │ │ │ │ ├── ProtonGPUTypes.td │ │ │ │ └── Types.h │ │ │ │ └── Transforms │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── Passes.h │ │ │ │ └── Passes.td │ │ ├── lib │ │ │ ├── Analysis │ │ │ │ ├── CMakeLists.txt │ │ │ │ └── ScopeIdAllocation.cpp │ │ │ ├── CMakeLists.txt │ │ │ ├── Dialect │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── Proton │ │ │ │ │ ├── CMakeLists.txt │ │ │ │ │ └── IR │ │ │ │ │ │ ├── CMakeLists.txt │ │ │ │ │ │ ├── Dialect.cpp │ │ │ │ │ │ └── Ops.cpp │ │ │ │ └── ProtonGPU │ │ │ │ │ ├── CMakeLists.txt │ │ │ │ │ ├── IR │ │ │ │ │ ├── CMakeLists.txt │ │ │ │ │ ├── Dialect.cpp │ │ │ │ │ ├── Ops.cpp │ │ │ │ │ └── Types.cpp │ │ │ │ │ └── Transforms │ │ │ │ │ ├── CMakeLists.txt │ │ │ │ │ └── ProtonGPUTransformsPass.cpp │ │ │ ├── ProtonGPUToLLVM │ │ │ │ ├── AllocateProtonGlobalScratchBuffer.cpp │ │ │ │ ├── AllocateProtonSharedMemory.cpp │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── PatternProtonGPUOpToLLVM.cpp │ │ │ │ ├── ProtonAMDGPUToLLVM │ │ │ │ │ ├── AMDPatternProtonGPUOpToLLVM.cpp │ │ │ │ │ ├── AddSchedBarriers.cpp │ │ │ │ │ ├── CMakeLists.txt │ │ │ │ │ ├── ConvertProtonGPUToLLVM.cpp │ │ │ │ │ └── TargetInfo.cpp │ │ │ │ ├── ProtonNvidiaGPUToLLVM │ │ │ │ │ ├── CMakeLists.txt │ │ │ │ │ ├── ConvertProtonGPUToLLVM.cpp │ │ │ │ │ ├── NvidiaPatternProtonGPUOpToLLVM.cpp │ │ │ │ │ └── TargetInfo.cpp │ │ │ │ └── Utility.cpp │ │ │ └── ProtonToProtonGPU │ │ │ │ ├── CMakeLists.txt │ │ │ │ └── ProtonToProtonGPUPass.cpp │ │ └── triton_proton.cc │ ├── README.md │ ├── common │ │ ├── CMakeLists.txt │ │ ├── include │ │ │ ├── Device.h │ │ │ └── TraceDataIO │ │ │ │ ├── ByteSpan.h │ │ │ │ ├── CircularLayoutParser.h │ │ │ │ ├── EntryDecoder.h │ │ │ │ ├── Parser.h │ │ │ │ └── TraceWriter.h │ │ └── lib │ │ │ ├── CMakeLists.txt │ │ │ └── TraceDataIO │ │ │ ├── ByteSpan.cpp │ │ │ ├── CMakeLists.txt │ │ │ ├── CircularLayoutParser.cpp │ │ │ ├── EntryDecoder.cpp │ │ │ ├── Parser.cpp │ │ │ └── TraceWriter.cpp │ ├── csrc │ │ ├── CMakeLists.txt │ │ ├── Proton.cpp │ │ ├── include │ │ │ ├── Context │ │ │ │ ├── Context.h │ │ │ │ ├── Python.h │ │ │ │ └── Shadow.h │ │ │ ├── Data │ │ │ │ ├── Data.h │ │ │ │ ├── Metric.h │ │ │ │ ├── TraceData.h │ │ │ │ └── TreeData.h │ │ │ ├── Driver │ │ │ │ ├── Dispatch.h │ │ │ │ └── GPU │ │ │ │ │ ├── CudaApi.h │ │ │ │ │ ├── CuptiApi.h │ │ │ │ │ ├── HipApi.h │ │ │ │ │ ├── HsaApi.h │ │ │ │ │ └── RoctracerApi.h │ │ │ ├── Profiler │ │ │ │ ├── Cupti │ │ │ │ │ ├── CuptiPCSampling.h │ │ │ │ │ └── CuptiProfiler.h │ │ │ │ ├── GPUProfiler.h │ │ │ │ ├── Instrumentation │ │ │ │ │ ├── CudaRuntime.h │ │ │ │ │ ├── HipRuntime.h │ │ │ │ │ ├── InstrumentationProfiler.h │ │ │ │ │ ├── Metadata.h │ │ │ │ │ └── Runtime.h │ │ │ │ ├── Profiler.h │ │ │ │ └── Roctracer │ │ │ │ │ └── RoctracerProfiler.h │ │ │ ├── Proton.h │ │ │ ├── Session │ │ │ │ └── Session.h │ │ │ └── Utility │ │ │ │ ├── Atomic.h │ │ │ │ ├── Errors.h │ │ │ │ ├── Map.h │ │ │ │ ├── Numeric.h │ │ │ │ ├── Set.h │ │ │ │ ├── Singleton.h │ │ │ │ ├── String.h │ │ │ │ └── Traits.h │ │ └── lib │ │ │ ├── CMakeLists.txt │ │ │ ├── Context │ │ │ ├── CMakeLists.txt │ │ │ ├── Context.cpp │ │ │ ├── Python.cpp │ │ │ └── Shadow.cpp │ │ │ ├── Data │ │ │ ├── CMakeLists.txt │ │ │ ├── Data.cpp │ │ │ ├── TraceData.cpp │ │ │ └── TreeData.cpp │ │ │ ├── Driver │ │ │ ├── CMakeLists.txt │ │ │ ├── Device.cpp │ │ │ └── GPU │ │ │ │ ├── CudaApi.cpp │ │ │ │ ├── CuptiApi.cpp │ │ │ │ ├── HipApi.cpp │ │ │ │ ├── HsaApi.cpp │ │ │ │ └── RoctracerApi.cpp │ │ │ ├── Profiler │ │ │ ├── CMakeLists.txt │ │ │ ├── Cupti │ │ │ │ ├── CuptiPCSampling.cpp │ │ │ │ └── CuptiProfiler.cpp │ │ │ ├── Instrumentation │ │ │ │ ├── CudaRuntime.cpp │ │ │ │ ├── HipRuntime.cpp │ │ │ │ ├── InstrumentationProfiler.cpp │ │ │ │ └── Metadata.cpp │ │ │ └── RocTracer │ │ │ │ └── RoctracerProfiler.cpp │ │ │ └── Session │ │ │ ├── CMakeLists.txt │ │ │ └── Session.cpp │ ├── proton │ │ ├── __init__.py │ │ ├── context.py │ │ ├── flags.py │ │ ├── hooks │ │ │ ├── __init__.py │ │ │ ├── hook.py │ │ │ ├── instrumentation.py │ │ │ └── launch.py │ │ ├── language.py │ │ ├── mode.py │ │ ├── profile.py │ │ ├── proton.py │ │ ├── scope.py │ │ ├── specs.py │ │ ├── state.py │ │ └── viewer.py │ ├── scripts │ │ └── dump_ttgir.sh │ ├── test │ │ ├── CMakeLists.txt │ │ ├── examples │ │ │ ├── cuda.json │ │ │ ├── frame.json │ │ │ ├── hip.json │ │ │ ├── leaf_nodes.json │ │ │ └── triton.json │ │ ├── helper.py │ │ ├── helper_kernels.py │ │ ├── override_helper.py │ │ ├── test_api.py │ │ ├── test_cmd.py │ │ ├── test_instrumentation.py │ │ ├── test_lib.py │ │ ├── test_override.py │ │ ├── test_profile.py │ │ ├── test_viewer.py │ │ └── unittest │ │ │ ├── CMakeLists.txt │ │ │ ├── TraceDataIO │ │ │ ├── ByteSpanTest.cpp │ │ │ ├── CMakeLists.txt │ │ │ ├── ChromeTraceWriterTest.cpp │ │ │ ├── CircularLayoutParserTest.cpp │ │ │ └── DecoderTest.cpp │ │ │ └── util │ │ │ ├── loop.bin │ │ │ ├── seq.bin │ │ │ └── trace_gen.py │ └── tutorials │ │ ├── dynamic-net.py │ │ ├── intra_kernel │ │ ├── README.md │ │ ├── example_dsl.py │ │ ├── example_override.py │ │ └── insert_proton_records │ │ └── matmul.py └── tlx │ ├── CMakeLists.txt │ ├── README.md │ ├── denoise.sh │ ├── dialect │ ├── CMakeLists.txt │ ├── include │ │ ├── Analysis │ │ │ └── LayoutPropagation.h │ │ ├── CMakeLists.txt │ │ ├── IR │ │ │ ├── CMakeLists.txt │ │ │ ├── Dialect.h │ │ │ ├── TLXAttrDefs.td │ │ │ ├── TLXDialect.td │ │ │ ├── TLXInterfaces.td │ │ │ ├── TLXOps.td │ │ │ └── Traits.h │ │ └── Transforms │ │ │ ├── CMakeLists.txt │ │ │ ├── Passes.h │ │ │ └── Passes.td │ ├── lib │ │ ├── Analysis │ │ │ ├── CMakeLists.txt │ │ │ └── LayoutPropagation.cpp │ │ ├── CMakeLists.txt │ │ ├── IR │ │ │ ├── CMakeLists.txt │ │ │ ├── Dialect.cpp │ │ │ ├── Ops.cpp │ │ │ └── Traits.cpp │ │ └── Transforms │ │ │ ├── CMakeLists.txt │ │ │ ├── Fixup.cpp │ │ │ ├── InsertRequireLayout.cpp │ │ │ ├── PropagateLayout.cpp │ │ │ └── RewriteLocalAlias.cpp │ └── triton_tlx.cc │ ├── doc │ ├── TLX-triton-conference.pdf │ └── tlx_barriers.md │ ├── language │ └── tlx │ │ ├── __init__.py │ │ ├── async_task_utils.py │ │ ├── barrier.py │ │ ├── compiler │ │ ├── __init__.py │ │ ├── code_generator.py │ │ └── dispatch.py │ │ ├── dynamic_launch.py │ │ ├── mem_ops.py │ │ ├── mma_ops.py │ │ ├── types.py │ │ └── utility.py │ ├── media │ ├── image1.PNG │ ├── image2.PNG │ ├── image3.PNG │ ├── image4.PNG │ └── image5.PNG │ ├── run_all.sh │ └── tutorials │ ├── .gitignore │ ├── amd-gemm-pipelined.py │ ├── blackwell-fa-ws-persistent_test.py │ ├── blackwell-fa-ws-pipelined-persistent_test.py │ ├── blackwell-fa-ws-pipelined_test.py │ ├── blackwell-fa-ws_test.py │ ├── blackwell-gdpa.py │ ├── blackwell-gemm-2cta_test.py │ ├── blackwell-gemm-clc_test.py │ ├── blackwell-gemm-pipelined_test.py │ ├── blackwell-gemm-ws-2cta.py │ ├── blackwell-gemm-ws_test.py │ ├── blackwell-grouped-gemm_test.py │ ├── hopper-fa-ws-pipelined-pingpong_test.py │ ├── hopper-fa-ws-pipelined_test.py │ ├── hopper-fa-ws_test.py │ ├── hopper-gemm-pipelined_test.py │ ├── hopper-gemm-ws_test.py │ ├── hopper-persistent-gemm-ws-cooperative.py │ ├── hopper-persistent-gemm-ws-pingpong.py │ └── vector-add2.py ├── unittest ├── Analysis │ ├── CMakeLists.txt │ └── UtilityTest.cpp ├── CMakeLists.txt ├── Dialect │ ├── CMakeLists.txt │ └── TritonGPU │ │ ├── CMakeLists.txt │ │ ├── DialectTest.cpp │ │ ├── DumpLayoutTest.cpp │ │ ├── LinearLayoutConversionsTest.cpp │ │ └── SwizzleTest.cpp ├── Tools │ ├── CMakeLists.txt │ ├── LayoutUtilsTest.cpp │ └── LinearLayoutTest.cpp └── googletest.cmake └── utils ├── generate-test-checks.py └── nightly.pypirc /.clang-format: -------------------------------------------------------------------------------- 1 | BasedOnStyle: LLVM 2 | -------------------------------------------------------------------------------- /.editorconfig: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/.editorconfig -------------------------------------------------------------------------------- /.git-blame-ignore-revs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/.git-blame-ignore-revs -------------------------------------------------------------------------------- /.github/CODEOWNERS: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/.github/CODEOWNERS -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/bug.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/.github/ISSUE_TEMPLATE/bug.yml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/.github/ISSUE_TEMPLATE/config.yml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/performance.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/.github/ISSUE_TEMPLATE/performance.yml -------------------------------------------------------------------------------- /.github/PULL_REQUEST_TEMPLATE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/.github/PULL_REQUEST_TEMPLATE.md -------------------------------------------------------------------------------- /.github/dependabot.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/.github/dependabot.yml -------------------------------------------------------------------------------- /.github/workflows/build-macos.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/.github/workflows/build-macos.yml -------------------------------------------------------------------------------- /.github/workflows/ci.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/.github/workflows/ci.yml -------------------------------------------------------------------------------- /.github/workflows/create_release.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/.github/workflows/create_release.yml -------------------------------------------------------------------------------- /.github/workflows/documentation.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/.github/workflows/documentation.yml -------------------------------------------------------------------------------- /.github/workflows/llvm-build.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/.github/workflows/llvm-build.yml -------------------------------------------------------------------------------- /.github/workflows/llvm-build/almalinux.Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/.github/workflows/llvm-build/almalinux.Dockerfile -------------------------------------------------------------------------------- /.github/workflows/llvm-build/centos.Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/.github/workflows/llvm-build/centos.Dockerfile -------------------------------------------------------------------------------- /.github/workflows/pre-commit.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/.github/workflows/pre-commit.yml -------------------------------------------------------------------------------- /.github/workflows/runner-preparation.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/.github/workflows/runner-preparation.yml -------------------------------------------------------------------------------- /.github/workflows/wheels.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/.github/workflows/wheels.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/CMakeLists.txt -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/README.md -------------------------------------------------------------------------------- /RELEASE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/RELEASE.md -------------------------------------------------------------------------------- /bin/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/bin/CMakeLists.txt -------------------------------------------------------------------------------- /bin/RegisterTritonDialects.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/bin/RegisterTritonDialects.h -------------------------------------------------------------------------------- /bin/triton-llvm-opt.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/bin/triton-llvm-opt.cpp -------------------------------------------------------------------------------- /bin/triton-lsp.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/bin/triton-lsp.cpp -------------------------------------------------------------------------------- /bin/triton-opt.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/bin/triton-opt.cpp -------------------------------------------------------------------------------- /bin/triton-reduce.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/bin/triton-reduce.cpp -------------------------------------------------------------------------------- /bin/triton-tensor-layout.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/bin/triton-tensor-layout.cpp -------------------------------------------------------------------------------- /cmake/AddTritonUnitTest.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/cmake/AddTritonUnitTest.cmake -------------------------------------------------------------------------------- /cmake/FindLLVM.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/cmake/FindLLVM.cmake -------------------------------------------------------------------------------- /cmake/json-version.txt: -------------------------------------------------------------------------------- 1 | v3.11.3 2 | -------------------------------------------------------------------------------- /cmake/llvm-hash.txt: -------------------------------------------------------------------------------- 1 | f6ded0be897e2878612dd903f7e8bb85448269e5 2 | -------------------------------------------------------------------------------- /cmake/nvidia-toolchain-version.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/cmake/nvidia-toolchain-version.json -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/_templates/versions.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/docs/_templates/versions.html -------------------------------------------------------------------------------- /docs/backend/ldmatrixOperand0.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/docs/backend/ldmatrixOperand0.svg -------------------------------------------------------------------------------- /docs/backend/ldmatrixOperand1.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/docs/backend/ldmatrixOperand1.svg -------------------------------------------------------------------------------- /docs/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/docs/conf.py -------------------------------------------------------------------------------- /docs/getting-started/installation.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/docs/getting-started/installation.rst -------------------------------------------------------------------------------- /docs/getting-started/tutorials/random_bits.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/docs/getting-started/tutorials/random_bits.png -------------------------------------------------------------------------------- /docs/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/docs/index.rst -------------------------------------------------------------------------------- /docs/meetups/01-24-2024/notes.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/docs/meetups/01-24-2024/notes.md -------------------------------------------------------------------------------- /docs/meetups/02-20-2024/Proton.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/docs/meetups/02-20-2024/Proton.pdf -------------------------------------------------------------------------------- /docs/meetups/02-20-2024/notes.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/docs/meetups/02-20-2024/notes.md -------------------------------------------------------------------------------- /docs/meetups/03-12-2025/notes.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/docs/meetups/03-12-2025/notes.md -------------------------------------------------------------------------------- /docs/meetups/04-02-2024/notes.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/docs/meetups/04-02-2024/notes.md -------------------------------------------------------------------------------- /docs/meetups/05-01-2025/notes.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/docs/meetups/05-01-2025/notes.md -------------------------------------------------------------------------------- /docs/meetups/05-07-2024/notes.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/docs/meetups/05-07-2024/notes.md -------------------------------------------------------------------------------- /docs/meetups/07-09-2025/notes.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/docs/meetups/07-09-2025/notes.md -------------------------------------------------------------------------------- /docs/meetups/07-18-2023/notes.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/docs/meetups/07-18-2023/notes.md -------------------------------------------------------------------------------- /docs/meetups/08-06-2024/notes.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/docs/meetups/08-06-2024/notes.md -------------------------------------------------------------------------------- /docs/meetups/08-22-2023/amd-update.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/docs/meetups/08-22-2023/amd-update.pdf -------------------------------------------------------------------------------- /docs/meetups/08-22-2023/intel-xpu-update.pptx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/docs/meetups/08-22-2023/intel-xpu-update.pptx -------------------------------------------------------------------------------- /docs/meetups/08-22-2023/notes.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/docs/meetups/08-22-2023/notes.md -------------------------------------------------------------------------------- /docs/meetups/10-25-2023/intel-xpu-update.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/docs/meetups/10-25-2023/intel-xpu-update.pdf -------------------------------------------------------------------------------- /docs/meetups/10-25-2023/notes.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/docs/meetups/10-25-2023/notes.md -------------------------------------------------------------------------------- /docs/meetups/10-25-2023/triton-shared.pptx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/docs/meetups/10-25-2023/triton-shared.pptx -------------------------------------------------------------------------------- /docs/meetups/12-13-2023/notes.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/docs/meetups/12-13-2023/notes.md -------------------------------------------------------------------------------- /docs/meetups/dev-meetup-2023.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/docs/meetups/dev-meetup-2023.md -------------------------------------------------------------------------------- /docs/meetups/dev_conference_2024.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/docs/meetups/dev_conference_2024.md -------------------------------------------------------------------------------- /docs/meetups/for_moderators/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/docs/meetups/for_moderators/README.md -------------------------------------------------------------------------------- /docs/programming-guide/chapter-1/introduction.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/docs/programming-guide/chapter-1/introduction.rst -------------------------------------------------------------------------------- /docs/programming-guide/chapter-2/related-work.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/docs/programming-guide/chapter-2/related-work.rst -------------------------------------------------------------------------------- /docs/programming-guide/chapter-3/debugging.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/docs/programming-guide/chapter-3/debugging.rst -------------------------------------------------------------------------------- /docs/python-api/triton-semantics.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/docs/python-api/triton-semantics.rst -------------------------------------------------------------------------------- /docs/python-api/triton.language.extra.cuda.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/docs/python-api/triton.language.extra.cuda.rst -------------------------------------------------------------------------------- /docs/python-api/triton.language.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/docs/python-api/triton.language.rst -------------------------------------------------------------------------------- /docs/python-api/triton.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/docs/python-api/triton.rst -------------------------------------------------------------------------------- /docs/python-api/triton.testing.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/docs/python-api/triton.testing.rst -------------------------------------------------------------------------------- /docs/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/docs/requirements.txt -------------------------------------------------------------------------------- /include/CMakeLists.txt: -------------------------------------------------------------------------------- 1 | add_subdirectory(triton) 2 | -------------------------------------------------------------------------------- /include/triton/Analysis/Alias.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Analysis/Alias.h -------------------------------------------------------------------------------- /include/triton/Analysis/Allocation.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Analysis/Allocation.h -------------------------------------------------------------------------------- /include/triton/Analysis/AxisInfo.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Analysis/AxisInfo.h -------------------------------------------------------------------------------- /include/triton/Analysis/Membar.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Analysis/Membar.h -------------------------------------------------------------------------------- /include/triton/Analysis/Utility.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Analysis/Utility.h -------------------------------------------------------------------------------- /include/triton/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/CMakeLists.txt -------------------------------------------------------------------------------- /include/triton/Conversion/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Conversion/CMakeLists.txt -------------------------------------------------------------------------------- /include/triton/Conversion/MLIRTypes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Conversion/MLIRTypes.h -------------------------------------------------------------------------------- /include/triton/Conversion/TritonGPUToLLVM/Passes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Conversion/TritonGPUToLLVM/Passes.h -------------------------------------------------------------------------------- /include/triton/Dialect/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Dialect/CMakeLists.txt -------------------------------------------------------------------------------- /include/triton/Dialect/Gluon/CMakeCache.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Dialect/Gluon/CMakeCache.txt -------------------------------------------------------------------------------- /include/triton/Dialect/Gluon/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Dialect/Gluon/CMakeLists.txt -------------------------------------------------------------------------------- /include/triton/Dialect/Gluon/IR/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Dialect/Gluon/IR/CMakeLists.txt -------------------------------------------------------------------------------- /include/triton/Dialect/Gluon/IR/Dialect.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Dialect/Gluon/IR/Dialect.h -------------------------------------------------------------------------------- /include/triton/Dialect/Gluon/IR/GluonAttrDefs.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Dialect/Gluon/IR/GluonAttrDefs.td -------------------------------------------------------------------------------- /include/triton/Dialect/Gluon/IR/GluonDialect.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Dialect/Gluon/IR/GluonDialect.td -------------------------------------------------------------------------------- /include/triton/Dialect/Gluon/IR/GluonOps.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Dialect/Gluon/IR/GluonOps.td -------------------------------------------------------------------------------- /include/triton/Dialect/Gluon/Transforms/Passes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Dialect/Gluon/Transforms/Passes.h -------------------------------------------------------------------------------- /include/triton/Dialect/Gluon/Transforms/Passes.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Dialect/Gluon/Transforms/Passes.td -------------------------------------------------------------------------------- /include/triton/Dialect/Triton/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Dialect/Triton/CMakeLists.txt -------------------------------------------------------------------------------- /include/triton/Dialect/Triton/IR/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Dialect/Triton/IR/CMakeLists.txt -------------------------------------------------------------------------------- /include/triton/Dialect/Triton/IR/Dialect.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Dialect/Triton/IR/Dialect.h -------------------------------------------------------------------------------- /include/triton/Dialect/Triton/IR/Interfaces.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Dialect/Triton/IR/Interfaces.h -------------------------------------------------------------------------------- /include/triton/Dialect/Triton/IR/OpInterfaces.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Dialect/Triton/IR/OpInterfaces.h -------------------------------------------------------------------------------- /include/triton/Dialect/Triton/IR/Traits.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Dialect/Triton/IR/Traits.h -------------------------------------------------------------------------------- /include/triton/Dialect/Triton/IR/TritonAttrDefs.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Dialect/Triton/IR/TritonAttrDefs.td -------------------------------------------------------------------------------- /include/triton/Dialect/Triton/IR/TritonDialect.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Dialect/Triton/IR/TritonDialect.td -------------------------------------------------------------------------------- /include/triton/Dialect/Triton/IR/TritonOps.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Dialect/Triton/IR/TritonOps.td -------------------------------------------------------------------------------- /include/triton/Dialect/Triton/IR/TritonTypes.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Dialect/Triton/IR/TritonTypes.td -------------------------------------------------------------------------------- /include/triton/Dialect/Triton/IR/Types.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Dialect/Triton/IR/Types.h -------------------------------------------------------------------------------- /include/triton/Dialect/Triton/IR/Utility.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Dialect/Triton/IR/Utility.h -------------------------------------------------------------------------------- /include/triton/Dialect/Triton/Transforms/Passes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Dialect/Triton/Transforms/Passes.h -------------------------------------------------------------------------------- /include/triton/Dialect/Triton/Transforms/Passes.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Dialect/Triton/Transforms/Passes.td -------------------------------------------------------------------------------- /include/triton/Dialect/TritonGPU/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Dialect/TritonGPU/CMakeLists.txt -------------------------------------------------------------------------------- /include/triton/Dialect/TritonGPU/IR/Attributes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Dialect/TritonGPU/IR/Attributes.h -------------------------------------------------------------------------------- /include/triton/Dialect/TritonGPU/IR/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Dialect/TritonGPU/IR/CMakeLists.txt -------------------------------------------------------------------------------- /include/triton/Dialect/TritonGPU/IR/Dialect.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Dialect/TritonGPU/IR/Dialect.h -------------------------------------------------------------------------------- /include/triton/Dialect/TritonGPU/IR/Traits.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Dialect/TritonGPU/IR/Traits.h -------------------------------------------------------------------------------- /include/triton/Dialect/TritonGPU/IR/Types.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Dialect/TritonGPU/IR/Types.h -------------------------------------------------------------------------------- /include/triton/Target/CMakeLists.txt: -------------------------------------------------------------------------------- 1 | add_subdirectory(LLVMIR) 2 | -------------------------------------------------------------------------------- /include/triton/Target/LLVMIR/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Target/LLVMIR/CMakeLists.txt -------------------------------------------------------------------------------- /include/triton/Target/LLVMIR/Passes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Target/LLVMIR/Passes.h -------------------------------------------------------------------------------- /include/triton/Target/LLVMIR/Passes.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Target/LLVMIR/Passes.td -------------------------------------------------------------------------------- /include/triton/Tools/GenericSwizzling.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Tools/GenericSwizzling.h -------------------------------------------------------------------------------- /include/triton/Tools/LayoutUtils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Tools/LayoutUtils.h -------------------------------------------------------------------------------- /include/triton/Tools/LinearLayout.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Tools/LinearLayout.h -------------------------------------------------------------------------------- /include/triton/Tools/StrUtil.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Tools/StrUtil.h -------------------------------------------------------------------------------- /include/triton/Tools/Sys/GetEnv.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/include/triton/Tools/Sys/GetEnv.hpp -------------------------------------------------------------------------------- /lib/Analysis/Alias.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Analysis/Alias.cpp -------------------------------------------------------------------------------- /lib/Analysis/Allocation.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Analysis/Allocation.cpp -------------------------------------------------------------------------------- /lib/Analysis/AxisInfo.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Analysis/AxisInfo.cpp -------------------------------------------------------------------------------- /lib/Analysis/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Analysis/CMakeLists.txt -------------------------------------------------------------------------------- /lib/Analysis/Membar.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Analysis/Membar.cpp -------------------------------------------------------------------------------- /lib/Analysis/Utility.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Analysis/Utility.cpp -------------------------------------------------------------------------------- /lib/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/CMakeLists.txt -------------------------------------------------------------------------------- /lib/Conversion/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Conversion/CMakeLists.txt -------------------------------------------------------------------------------- /lib/Conversion/TritonGPUToLLVM/AssertOpToLLVM.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Conversion/TritonGPUToLLVM/AssertOpToLLVM.cpp -------------------------------------------------------------------------------- /lib/Conversion/TritonGPUToLLVM/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Conversion/TritonGPUToLLVM/CMakeLists.txt -------------------------------------------------------------------------------- /lib/Conversion/TritonGPUToLLVM/DotOpToLLVM/FMA.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Conversion/TritonGPUToLLVM/DotOpToLLVM/FMA.cpp -------------------------------------------------------------------------------- /lib/Conversion/TritonGPUToLLVM/FuncOpToLLVM.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Conversion/TritonGPUToLLVM/FuncOpToLLVM.cpp -------------------------------------------------------------------------------- /lib/Conversion/TritonGPUToLLVM/GatherOpToLLVM.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Conversion/TritonGPUToLLVM/GatherOpToLLVM.cpp -------------------------------------------------------------------------------- /lib/Conversion/TritonGPUToLLVM/MemoryOpToLLVM.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Conversion/TritonGPUToLLVM/MemoryOpToLLVM.cpp -------------------------------------------------------------------------------- /lib/Conversion/TritonGPUToLLVM/PrintOpToLLVM.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Conversion/TritonGPUToLLVM/PrintOpToLLVM.cpp -------------------------------------------------------------------------------- /lib/Conversion/TritonGPUToLLVM/ReduceOpToLLVM.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Conversion/TritonGPUToLLVM/ReduceOpToLLVM.cpp -------------------------------------------------------------------------------- /lib/Conversion/TritonGPUToLLVM/ReduceScanCommon.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Conversion/TritonGPUToLLVM/ReduceScanCommon.h -------------------------------------------------------------------------------- /lib/Conversion/TritonGPUToLLVM/SPMDOpToLLVM.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Conversion/TritonGPUToLLVM/SPMDOpToLLVM.cpp -------------------------------------------------------------------------------- /lib/Conversion/TritonGPUToLLVM/ScanOpToLLVM.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Conversion/TritonGPUToLLVM/ScanOpToLLVM.cpp -------------------------------------------------------------------------------- /lib/Conversion/TritonGPUToLLVM/TypeConverter.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Conversion/TritonGPUToLLVM/TypeConverter.cpp -------------------------------------------------------------------------------- /lib/Conversion/TritonGPUToLLVM/Utility.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Conversion/TritonGPUToLLVM/Utility.cpp -------------------------------------------------------------------------------- /lib/Conversion/TritonGPUToLLVM/ViewOpToLLVM.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Conversion/TritonGPUToLLVM/ViewOpToLLVM.cpp -------------------------------------------------------------------------------- /lib/Conversion/TritonToTritonGPU/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Conversion/TritonToTritonGPU/CMakeLists.txt -------------------------------------------------------------------------------- /lib/Dialect/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/CMakeLists.txt -------------------------------------------------------------------------------- /lib/Dialect/Gluon/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/Gluon/CMakeLists.txt -------------------------------------------------------------------------------- /lib/Dialect/Gluon/IR/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/Gluon/IR/CMakeLists.txt -------------------------------------------------------------------------------- /lib/Dialect/Gluon/IR/Dialect.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/Gluon/IR/Dialect.cpp -------------------------------------------------------------------------------- /lib/Dialect/Gluon/Transforms/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/Gluon/Transforms/CMakeLists.txt -------------------------------------------------------------------------------- /lib/Dialect/Gluon/Transforms/Canonicalize.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/Gluon/Transforms/Canonicalize.cpp -------------------------------------------------------------------------------- /lib/Dialect/Gluon/Transforms/Inline.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/Gluon/Transforms/Inline.cpp -------------------------------------------------------------------------------- /lib/Dialect/Triton/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/Triton/CMakeLists.txt -------------------------------------------------------------------------------- /lib/Dialect/Triton/IR/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/Triton/IR/CMakeLists.txt -------------------------------------------------------------------------------- /lib/Dialect/Triton/IR/Canonicalize.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/Triton/IR/Canonicalize.td -------------------------------------------------------------------------------- /lib/Dialect/Triton/IR/Dialect.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/Triton/IR/Dialect.cpp -------------------------------------------------------------------------------- /lib/Dialect/Triton/IR/DiscardableAttributes.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/Triton/IR/DiscardableAttributes.cpp -------------------------------------------------------------------------------- /lib/Dialect/Triton/IR/OpInterfaces.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/Triton/IR/OpInterfaces.cpp -------------------------------------------------------------------------------- /lib/Dialect/Triton/IR/Ops.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/Triton/IR/Ops.cpp -------------------------------------------------------------------------------- /lib/Dialect/Triton/IR/Traits.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/Triton/IR/Traits.cpp -------------------------------------------------------------------------------- /lib/Dialect/Triton/IR/Types.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/Triton/IR/Types.cpp -------------------------------------------------------------------------------- /lib/Dialect/Triton/IR/Utility.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/Triton/IR/Utility.cpp -------------------------------------------------------------------------------- /lib/Dialect/Triton/Transforms/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/Triton/Transforms/CMakeLists.txt -------------------------------------------------------------------------------- /lib/Dialect/Triton/Transforms/Combine.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/Triton/Transforms/Combine.cpp -------------------------------------------------------------------------------- /lib/Dialect/Triton/Transforms/Combine.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/Triton/Transforms/Combine.td -------------------------------------------------------------------------------- /lib/Dialect/Triton/Transforms/LoopAwareCSE.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/Triton/Transforms/LoopAwareCSE.cpp -------------------------------------------------------------------------------- /lib/Dialect/Triton/Transforms/LoopPeeling.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/Triton/Transforms/LoopPeeling.cpp -------------------------------------------------------------------------------- /lib/Dialect/Triton/Transforms/LoopUnroll.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/Triton/Transforms/LoopUnroll.cpp -------------------------------------------------------------------------------- /lib/Dialect/Triton/Transforms/ReorderBroadcast.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/Triton/Transforms/ReorderBroadcast.cpp -------------------------------------------------------------------------------- /lib/Dialect/TritonGPU/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/TritonGPU/CMakeLists.txt -------------------------------------------------------------------------------- /lib/Dialect/TritonGPU/IR/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/TritonGPU/IR/CMakeLists.txt -------------------------------------------------------------------------------- /lib/Dialect/TritonGPU/IR/Dialect.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/TritonGPU/IR/Dialect.cpp -------------------------------------------------------------------------------- /lib/Dialect/TritonGPU/IR/LayoutUtility.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/TritonGPU/IR/LayoutUtility.cpp -------------------------------------------------------------------------------- /lib/Dialect/TritonGPU/IR/Ops.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/TritonGPU/IR/Ops.cpp -------------------------------------------------------------------------------- /lib/Dialect/TritonGPU/IR/Types.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/TritonGPU/IR/Types.cpp -------------------------------------------------------------------------------- /lib/Dialect/TritonGPU/Transforms/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/TritonGPU/Transforms/CMakeLists.txt -------------------------------------------------------------------------------- /lib/Dialect/TritonGPU/Transforms/Coalesce.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/TritonGPU/Transforms/Coalesce.cpp -------------------------------------------------------------------------------- /lib/Dialect/TritonGPU/Transforms/F32DotTC.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/TritonGPU/Transforms/F32DotTC.cpp -------------------------------------------------------------------------------- /lib/Dialect/TritonGPU/Transforms/Prefetch.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/TritonGPU/Transforms/Prefetch.cpp -------------------------------------------------------------------------------- /lib/Dialect/TritonGPU/Transforms/Utility.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/TritonGPU/Transforms/Utility.cpp -------------------------------------------------------------------------------- /lib/Dialect/TritonInstrument/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/TritonInstrument/CMakeLists.txt -------------------------------------------------------------------------------- /lib/Dialect/TritonInstrument/IR/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/TritonInstrument/IR/CMakeLists.txt -------------------------------------------------------------------------------- /lib/Dialect/TritonInstrument/IR/Dialect.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/TritonInstrument/IR/Dialect.cpp -------------------------------------------------------------------------------- /lib/Dialect/TritonInstrument/IR/Ops.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/TritonInstrument/IR/Ops.cpp -------------------------------------------------------------------------------- /lib/Dialect/TritonInstrument/IR/Utility.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/TritonInstrument/IR/Utility.cpp -------------------------------------------------------------------------------- /lib/Dialect/TritonNvidiaGPU/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/TritonNvidiaGPU/CMakeLists.txt -------------------------------------------------------------------------------- /lib/Dialect/TritonNvidiaGPU/IR/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/TritonNvidiaGPU/IR/CMakeLists.txt -------------------------------------------------------------------------------- /lib/Dialect/TritonNvidiaGPU/IR/Dialect.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/TritonNvidiaGPU/IR/Dialect.cpp -------------------------------------------------------------------------------- /lib/Dialect/TritonNvidiaGPU/IR/Ops.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/TritonNvidiaGPU/IR/Ops.cpp -------------------------------------------------------------------------------- /lib/Dialect/TritonNvidiaGPU/Transforms/PlanCTA.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/TritonNvidiaGPU/Transforms/PlanCTA.cpp -------------------------------------------------------------------------------- /lib/Dialect/TritonNvidiaGPU/Transforms/Utility.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Dialect/TritonNvidiaGPU/Transforms/Utility.cpp -------------------------------------------------------------------------------- /lib/Instrumentation/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Instrumentation/CMakeLists.txt -------------------------------------------------------------------------------- /lib/Instrumentation/PrintLoadStoreMemSpaces.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Instrumentation/PrintLoadStoreMemSpaces.cpp -------------------------------------------------------------------------------- /lib/Target/CMakeLists.txt: -------------------------------------------------------------------------------- 1 | add_subdirectory(LLVMIR) 2 | -------------------------------------------------------------------------------- /lib/Target/LLVMIR/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Target/LLVMIR/CMakeLists.txt -------------------------------------------------------------------------------- /lib/Target/LLVMIR/LLVMDIScope.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Target/LLVMIR/LLVMDIScope.cpp -------------------------------------------------------------------------------- /lib/Target/LLVMIR/LLVMIRBreakPhiStruct.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Target/LLVMIR/LLVMIRBreakPhiStruct.cpp -------------------------------------------------------------------------------- /lib/Target/LLVMIR/LLVMPasses.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Target/LLVMIR/LLVMPasses.h -------------------------------------------------------------------------------- /lib/Tools/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Tools/CMakeLists.txt -------------------------------------------------------------------------------- /lib/Tools/GenericSwizzling.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Tools/GenericSwizzling.cpp -------------------------------------------------------------------------------- /lib/Tools/LayoutUtils.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Tools/LayoutUtils.cpp -------------------------------------------------------------------------------- /lib/Tools/LinearLayout.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/lib/Tools/LinearLayout.cpp -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/pyproject.toml -------------------------------------------------------------------------------- /python/build_helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/build_helpers.py -------------------------------------------------------------------------------- /python/examples/gluon/01-attention-forward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/examples/gluon/01-attention-forward.py -------------------------------------------------------------------------------- /python/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/requirements.txt -------------------------------------------------------------------------------- /python/src/gluon_ir.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/src/gluon_ir.cc -------------------------------------------------------------------------------- /python/src/interpreter.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/src/interpreter.cc -------------------------------------------------------------------------------- /python/src/ir.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/src/ir.cc -------------------------------------------------------------------------------- /python/src/ir.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/src/ir.h -------------------------------------------------------------------------------- /python/src/llvm.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/src/llvm.cc -------------------------------------------------------------------------------- /python/src/main.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/src/main.cc -------------------------------------------------------------------------------- /python/src/passes.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/src/passes.cc -------------------------------------------------------------------------------- /python/src/passes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/src/passes.h -------------------------------------------------------------------------------- /python/test-requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test-requirements.txt -------------------------------------------------------------------------------- /python/test/backend/extension_backend.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/backend/extension_backend.c -------------------------------------------------------------------------------- /python/test/backend/test_device_backend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/backend/test_device_backend.py -------------------------------------------------------------------------------- /python/test/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/conftest.py -------------------------------------------------------------------------------- /python/test/gluon/test_consan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/gluon/test_consan.py -------------------------------------------------------------------------------- /python/test/gluon/test_core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/gluon/test_core.py -------------------------------------------------------------------------------- /python/test/gluon/test_frontend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/gluon/test_frontend.py -------------------------------------------------------------------------------- /python/test/gluon/test_lowerings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/gluon/test_lowerings.py -------------------------------------------------------------------------------- /python/test/kernel_comparison/kernels.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/kernel_comparison/kernels.yml -------------------------------------------------------------------------------- /python/test/microbenchmark/launch_overhead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/microbenchmark/launch_overhead.py -------------------------------------------------------------------------------- /python/test/regression/test_cast_matmul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/regression/test_cast_matmul.py -------------------------------------------------------------------------------- /python/test/unit/cuda/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /python/test/unit/cuda/test_experimental_tma.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/cuda/test_experimental_tma.py -------------------------------------------------------------------------------- /python/test/unit/cuda/test_mixed_io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/cuda/test_mixed_io.py -------------------------------------------------------------------------------- /python/test/unit/cuda/test_tensor_descriptor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/cuda/test_tensor_descriptor.py -------------------------------------------------------------------------------- /python/test/unit/cuda/test_tma_descriptor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/cuda/test_tma_descriptor.py -------------------------------------------------------------------------------- /python/test/unit/cuda/test_tma_store_gemm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/cuda/test_tma_store_gemm.py -------------------------------------------------------------------------------- /python/test/unit/instrumentation/test_gpuhello.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/instrumentation/test_gpuhello.py -------------------------------------------------------------------------------- /python/test/unit/language/print_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/language/print_helper.py -------------------------------------------------------------------------------- /python/test/unit/language/test_annotations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/language/test_annotations.py -------------------------------------------------------------------------------- /python/test/unit/language/test_block_pointer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/language/test_block_pointer.py -------------------------------------------------------------------------------- /python/test/unit/language/test_compile_errors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/language/test_compile_errors.py -------------------------------------------------------------------------------- /python/test/unit/language/test_compile_only.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/language/test_compile_only.py -------------------------------------------------------------------------------- /python/test/unit/language/test_conversions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/language/test_conversions.py -------------------------------------------------------------------------------- /python/test/unit/language/test_core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/language/test_core.py -------------------------------------------------------------------------------- /python/test/unit/language/test_decorator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/language/test_decorator.py -------------------------------------------------------------------------------- /python/test/unit/language/test_frontend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/language/test_frontend.py -------------------------------------------------------------------------------- /python/test/unit/language/test_libdevice.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/language/test_libdevice.py -------------------------------------------------------------------------------- /python/test/unit/language/test_line_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/language/test_line_info.py -------------------------------------------------------------------------------- /python/test/unit/language/test_matmul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/language/test_matmul.py -------------------------------------------------------------------------------- /python/test/unit/language/test_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/language/test_module.py -------------------------------------------------------------------------------- /python/test/unit/language/test_mxfp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/language/test_mxfp.py -------------------------------------------------------------------------------- /python/test/unit/language/test_pipeliner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/language/test_pipeliner.py -------------------------------------------------------------------------------- /python/test/unit/language/test_random.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/language/test_random.py -------------------------------------------------------------------------------- /python/test/unit/language/test_reproducer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/language/test_reproducer.py -------------------------------------------------------------------------------- /python/test/unit/language/test_standard.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/language/test_standard.py -------------------------------------------------------------------------------- /python/test/unit/language/test_subprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/language/test_subprocess.py -------------------------------------------------------------------------------- /python/test/unit/language/test_tlx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/language/test_tlx.py -------------------------------------------------------------------------------- /python/test/unit/language/test_tuple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/language/test_tuple.py -------------------------------------------------------------------------------- /python/test/unit/runtime/test_autotuner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/runtime/test_autotuner.py -------------------------------------------------------------------------------- /python/test/unit/runtime/test_bindings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/runtime/test_bindings.py -------------------------------------------------------------------------------- /python/test/unit/runtime/test_build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/runtime/test_build.py -------------------------------------------------------------------------------- /python/test/unit/runtime/test_cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/runtime/test_cache.py -------------------------------------------------------------------------------- /python/test/unit/runtime/test_cublas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/runtime/test_cublas.py -------------------------------------------------------------------------------- /python/test/unit/runtime/test_driver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/runtime/test_driver.py -------------------------------------------------------------------------------- /python/test/unit/runtime/test_jit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/runtime/test_jit.py -------------------------------------------------------------------------------- /python/test/unit/runtime/test_launch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/runtime/test_launch.py -------------------------------------------------------------------------------- /python/test/unit/runtime/test_subproc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/runtime/test_subproc.py -------------------------------------------------------------------------------- /python/test/unit/test_debug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/test_debug.py -------------------------------------------------------------------------------- /python/test/unit/test_debug_dump.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/test_debug_dump.py -------------------------------------------------------------------------------- /python/test/unit/test_filecheck.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/test_filecheck.py -------------------------------------------------------------------------------- /python/test/unit/test_knobs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/test_knobs.py -------------------------------------------------------------------------------- /python/test/unit/test_link.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/test_link.py -------------------------------------------------------------------------------- /python/test/unit/test_perf_warning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/test_perf_warning.py -------------------------------------------------------------------------------- /python/test/unit/tools/test_aot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/tools/test_aot.py -------------------------------------------------------------------------------- /python/test/unit/tools/test_disasm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/tools/test_disasm.py -------------------------------------------------------------------------------- /python/test/unit/tools/test_irsource.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/test/unit/tools/test_irsource.py -------------------------------------------------------------------------------- /python/triton/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/__init__.py -------------------------------------------------------------------------------- /python/triton/_filecheck.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/_filecheck.py -------------------------------------------------------------------------------- /python/triton/_internal_testing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/_internal_testing.py -------------------------------------------------------------------------------- /python/triton/_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/_utils.py -------------------------------------------------------------------------------- /python/triton/backends/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/backends/__init__.py -------------------------------------------------------------------------------- /python/triton/backends/compiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/backends/compiler.py -------------------------------------------------------------------------------- /python/triton/backends/driver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/backends/driver.py -------------------------------------------------------------------------------- /python/triton/compiler/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/compiler/__init__.py -------------------------------------------------------------------------------- /python/triton/compiler/code_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/compiler/code_generator.py -------------------------------------------------------------------------------- /python/triton/compiler/compiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/compiler/compiler.py -------------------------------------------------------------------------------- /python/triton/compiler/errors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/compiler/errors.py -------------------------------------------------------------------------------- /python/triton/compiler/make_launcher.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /python/triton/errors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/errors.py -------------------------------------------------------------------------------- /python/triton/experimental/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /python/triton/experimental/gluon/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/experimental/gluon/__init__.py -------------------------------------------------------------------------------- /python/triton/experimental/gluon/_compiler.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /python/triton/experimental/gluon/_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/experimental/gluon/_runtime.py -------------------------------------------------------------------------------- /python/triton/experimental/gluon/language/_core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/experimental/gluon/language/_core.py -------------------------------------------------------------------------------- /python/triton/experimental/gluon/language/_math.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/experimental/gluon/language/_math.py -------------------------------------------------------------------------------- /python/triton/experimental/gluon/nvidia/hopper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/experimental/gluon/nvidia/hopper.py -------------------------------------------------------------------------------- /python/triton/knobs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/knobs.py -------------------------------------------------------------------------------- /python/triton/language/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/language/__init__.py -------------------------------------------------------------------------------- /python/triton/language/core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/language/core.py -------------------------------------------------------------------------------- /python/triton/language/extra/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/language/extra/__init__.py -------------------------------------------------------------------------------- /python/triton/language/extra/libdevice.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/language/extra/libdevice.py -------------------------------------------------------------------------------- /python/triton/language/math.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/language/math.py -------------------------------------------------------------------------------- /python/triton/language/random.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/language/random.py -------------------------------------------------------------------------------- /python/triton/language/semantic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/language/semantic.py -------------------------------------------------------------------------------- /python/triton/language/standard.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/language/standard.py -------------------------------------------------------------------------------- /python/triton/language/target_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/language/target_info.py -------------------------------------------------------------------------------- /python/triton/runtime/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/runtime/__init__.py -------------------------------------------------------------------------------- /python/triton/runtime/_allocation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/runtime/_allocation.py -------------------------------------------------------------------------------- /python/triton/runtime/_async_compile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/runtime/_async_compile.py -------------------------------------------------------------------------------- /python/triton/runtime/autotuner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/runtime/autotuner.py -------------------------------------------------------------------------------- /python/triton/runtime/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/runtime/build.py -------------------------------------------------------------------------------- /python/triton/runtime/cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/runtime/cache.py -------------------------------------------------------------------------------- /python/triton/runtime/driver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/runtime/driver.py -------------------------------------------------------------------------------- /python/triton/runtime/errors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/runtime/errors.py -------------------------------------------------------------------------------- /python/triton/runtime/interpreter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/runtime/interpreter.py -------------------------------------------------------------------------------- /python/triton/runtime/jit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/runtime/jit.py -------------------------------------------------------------------------------- /python/triton/testing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/testing.py -------------------------------------------------------------------------------- /python/triton/tools/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /python/triton/tools/build_extern.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/tools/build_extern.py -------------------------------------------------------------------------------- /python/triton/tools/compile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/tools/compile.py -------------------------------------------------------------------------------- /python/triton/tools/disasm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/tools/disasm.py -------------------------------------------------------------------------------- /python/triton/tools/experimental_descriptor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/tools/experimental_descriptor.py -------------------------------------------------------------------------------- /python/triton/tools/link.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/tools/link.py -------------------------------------------------------------------------------- /python/triton/tools/mxfp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/tools/mxfp.py -------------------------------------------------------------------------------- /python/triton/tools/ragged_tma.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/tools/ragged_tma.py -------------------------------------------------------------------------------- /python/triton/tools/tensor_descriptor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton/tools/tensor_descriptor.py -------------------------------------------------------------------------------- /python/triton_kernels/.gitignore: -------------------------------------------------------------------------------- 1 | triton_bench.egg-info/ 2 | -------------------------------------------------------------------------------- /python/triton_kernels/bench/bench_mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton_kernels/bench/bench_mlp.py -------------------------------------------------------------------------------- /python/triton_kernels/bench/bench_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton_kernels/bench/bench_utils.py -------------------------------------------------------------------------------- /python/triton_kernels/bench/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton_kernels/bench/distributed.py -------------------------------------------------------------------------------- /python/triton_kernels/pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton_kernels/pyproject.toml -------------------------------------------------------------------------------- /python/triton_kernels/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /python/triton_kernels/tests/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton_kernels/tests/conftest.py -------------------------------------------------------------------------------- /python/triton_kernels/tests/test_compaction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton_kernels/tests/test_compaction.py -------------------------------------------------------------------------------- /python/triton_kernels/tests/test_matmul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton_kernels/tests/test_matmul.py -------------------------------------------------------------------------------- /python/triton_kernels/tests/test_mxfp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton_kernels/tests/test_mxfp.py -------------------------------------------------------------------------------- /python/triton_kernels/tests/test_routing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton_kernels/tests/test_routing.py -------------------------------------------------------------------------------- /python/triton_kernels/tests/test_specialize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton_kernels/tests/test_specialize.py -------------------------------------------------------------------------------- /python/triton_kernels/tests/test_swiglu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton_kernels/tests/test_swiglu.py -------------------------------------------------------------------------------- /python/triton_kernels/tests/test_tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton_kernels/tests/test_tensor.py -------------------------------------------------------------------------------- /python/triton_kernels/triton_kernels/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton_kernels/triton_kernels/__init__.py -------------------------------------------------------------------------------- /python/triton_kernels/triton_kernels/compaction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton_kernels/triton_kernels/compaction.py -------------------------------------------------------------------------------- /python/triton_kernels/triton_kernels/matmul_ogs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton_kernels/triton_kernels/matmul_ogs.py -------------------------------------------------------------------------------- /python/triton_kernels/triton_kernels/numerics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton_kernels/triton_kernels/numerics.py -------------------------------------------------------------------------------- /python/triton_kernels/triton_kernels/numerics_details/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /python/triton_kernels/triton_kernels/roofline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton_kernels/triton_kernels/roofline.py -------------------------------------------------------------------------------- /python/triton_kernels/triton_kernels/routing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton_kernels/triton_kernels/routing.py -------------------------------------------------------------------------------- /python/triton_kernels/triton_kernels/specialize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton_kernels/triton_kernels/specialize.py -------------------------------------------------------------------------------- /python/triton_kernels/triton_kernels/swiglu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton_kernels/triton_kernels/swiglu.py -------------------------------------------------------------------------------- /python/triton_kernels/triton_kernels/tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton_kernels/triton_kernels/tensor.py -------------------------------------------------------------------------------- /python/triton_kernels/triton_kernels/testing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton_kernels/triton_kernels/testing.py -------------------------------------------------------------------------------- /python/triton_kernels/triton_kernels/topk.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/triton_kernels/triton_kernels/topk.py -------------------------------------------------------------------------------- /python/triton_kernels/triton_kernels/topk_details/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /python/tutorials/01-vector-add.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/tutorials/01-vector-add.py -------------------------------------------------------------------------------- /python/tutorials/02-fused-softmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/tutorials/02-fused-softmax.py -------------------------------------------------------------------------------- /python/tutorials/03-matrix-multiplication.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/tutorials/03-matrix-multiplication.py -------------------------------------------------------------------------------- /python/tutorials/04-low-memory-dropout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/tutorials/04-low-memory-dropout.py -------------------------------------------------------------------------------- /python/tutorials/05-layer-norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/tutorials/05-layer-norm.py -------------------------------------------------------------------------------- /python/tutorials/06-fused-attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/tutorials/06-fused-attention.py -------------------------------------------------------------------------------- /python/tutorials/07-extern-functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/tutorials/07-extern-functions.py -------------------------------------------------------------------------------- /python/tutorials/08-grouped-gemm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/tutorials/08-grouped-gemm.py -------------------------------------------------------------------------------- /python/tutorials/09-persistent-matmul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/tutorials/09-persistent-matmul.py -------------------------------------------------------------------------------- /python/tutorials/10-block-scaled-matmul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/tutorials/10-block-scaled-matmul.py -------------------------------------------------------------------------------- /python/tutorials/README.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/tutorials/README.rst -------------------------------------------------------------------------------- /python/tutorials/fused-attention-ws-device-tma.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/tutorials/fused-attention-ws-device-tma.py -------------------------------------------------------------------------------- /python/tutorials/fused-attention-ws.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/tutorials/fused-attention-ws.py -------------------------------------------------------------------------------- /python/tutorials/gluon/01-intro.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/tutorials/gluon/01-intro.py -------------------------------------------------------------------------------- /python/tutorials/gluon/02-layouts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/tutorials/gluon/02-layouts.py -------------------------------------------------------------------------------- /python/tutorials/gluon/03-async-copy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/tutorials/gluon/03-async-copy.py -------------------------------------------------------------------------------- /python/tutorials/gluon/04-tma.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/tutorials/gluon/04-tma.py -------------------------------------------------------------------------------- /python/tutorials/gluon/05-wgmma.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/tutorials/gluon/05-wgmma.py -------------------------------------------------------------------------------- /python/tutorials/gluon/06-tcgen05.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/tutorials/gluon/06-tcgen05.py -------------------------------------------------------------------------------- /python/tutorials/gluon/07-persistence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/tutorials/gluon/07-persistence.py -------------------------------------------------------------------------------- /python/tutorials/gluon/08-warp-specialization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/python/tutorials/gluon/08-warp-specialization.py -------------------------------------------------------------------------------- /scripts/build-llvm-project.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/scripts/build-llvm-project.sh -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/setup.py -------------------------------------------------------------------------------- /test/Analysis/amd/test-alignment.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Analysis/amd/test-alignment.mlir -------------------------------------------------------------------------------- /test/Analysis/test-alias.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Analysis/test-alias.mlir -------------------------------------------------------------------------------- /test/Analysis/test-alignment.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Analysis/test-alignment.mlir -------------------------------------------------------------------------------- /test/Analysis/test-allocation.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Analysis/test-allocation.mlir -------------------------------------------------------------------------------- /test/Analysis/test-membar-ttng.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Analysis/test-membar-ttng.mlir -------------------------------------------------------------------------------- /test/Analysis/test-membar.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Analysis/test-membar.mlir -------------------------------------------------------------------------------- /test/Analysis/test-transpose-axisinfo.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Analysis/test-transpose-axisinfo.mlir -------------------------------------------------------------------------------- /test/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/CMakeLists.txt -------------------------------------------------------------------------------- /test/Conversion/allocate_shared_memory.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/allocate_shared_memory.mlir -------------------------------------------------------------------------------- /test/Conversion/allocate_warp_groups.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/allocate_warp_groups.mlir -------------------------------------------------------------------------------- /test/Conversion/amd/allocate_shared_memory.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/amd/allocate_shared_memory.mlir -------------------------------------------------------------------------------- /test/Conversion/amd/amdgpu_membar.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/amd/amdgpu_membar.mlir -------------------------------------------------------------------------------- /test/Conversion/amd/async-ops-alias-scopes.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/amd/async-ops-alias-scopes.mlir -------------------------------------------------------------------------------- /test/Conversion/amd/async_ops_to_llvm.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/amd/async_ops_to_llvm.mlir -------------------------------------------------------------------------------- /test/Conversion/amd/async_ops_to_llvm_invalid.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/amd/async_ops_to_llvm_invalid.mlir -------------------------------------------------------------------------------- /test/Conversion/amd/buffer_atomic_cas.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/amd/buffer_atomic_cas.mlir -------------------------------------------------------------------------------- /test/Conversion/amd/buffer_load_store.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/amd/buffer_load_store.mlir -------------------------------------------------------------------------------- /test/Conversion/amd/builtin_func_to_llvm.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/amd/builtin_func_to_llvm.mlir -------------------------------------------------------------------------------- /test/Conversion/amd/compute-base-ptr.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/amd/compute-base-ptr.mlir -------------------------------------------------------------------------------- /test/Conversion/amd/convert_layout.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/amd/convert_layout.mlir -------------------------------------------------------------------------------- /test/Conversion/amd/dedup-by-constancy.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/amd/dedup-by-constancy.mlir -------------------------------------------------------------------------------- /test/Conversion/amd/ds_transpose.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/amd/ds_transpose.mlir -------------------------------------------------------------------------------- /test/Conversion/amd/fp_to_fp.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/amd/fp_to_fp.mlir -------------------------------------------------------------------------------- /test/Conversion/amd/in_thread_transpose.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/amd/in_thread_transpose.mlir -------------------------------------------------------------------------------- /test/Conversion/amd/invalid_concat_op.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/amd/invalid_concat_op.mlir -------------------------------------------------------------------------------- /test/Conversion/amd/load_store.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/amd/load_store.mlir -------------------------------------------------------------------------------- /test/Conversion/amd/math-denorm-handling.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/amd/math-denorm-handling.mlir -------------------------------------------------------------------------------- /test/Conversion/amd/mfma-shortcut.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/amd/mfma-shortcut.mlir -------------------------------------------------------------------------------- /test/Conversion/amd/minmax.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/amd/minmax.mlir -------------------------------------------------------------------------------- /test/Conversion/amd/tritongpu_to_llvm.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/amd/tritongpu_to_llvm.mlir -------------------------------------------------------------------------------- /test/Conversion/amd/tritongpu_to_llvm_rdna.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/amd/tritongpu_to_llvm_rdna.mlir -------------------------------------------------------------------------------- /test/Conversion/amd/upcast_mxfp.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/amd/upcast_mxfp.mlir -------------------------------------------------------------------------------- /test/Conversion/amd/wmma-v1-shortcut.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/amd/wmma-v1-shortcut.mlir -------------------------------------------------------------------------------- /test/Conversion/amd/wmma-v2-shortcut.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/amd/wmma-v2-shortcut.mlir -------------------------------------------------------------------------------- /test/Conversion/atomic_ldst.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/atomic_ldst.mlir -------------------------------------------------------------------------------- /test/Conversion/cvt_to_llvm.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/cvt_to_llvm.mlir -------------------------------------------------------------------------------- /test/Conversion/dedup-by-constancy.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/dedup-by-constancy.mlir -------------------------------------------------------------------------------- /test/Conversion/divide-by-0.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/divide-by-0.mlir -------------------------------------------------------------------------------- /test/Conversion/nvgpu_to_llvm.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/nvgpu_to_llvm.mlir -------------------------------------------------------------------------------- /test/Conversion/reduce_to_llvm.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/reduce_to_llvm.mlir -------------------------------------------------------------------------------- /test/Conversion/relayout_tritongpu.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/relayout_tritongpu.mlir -------------------------------------------------------------------------------- /test/Conversion/scan_to_llvm.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/scan_to_llvm.mlir -------------------------------------------------------------------------------- /test/Conversion/tma_to_llvm.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/tma_to_llvm.mlir -------------------------------------------------------------------------------- /test/Conversion/triton_to_tritongpu.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/triton_to_tritongpu.mlir -------------------------------------------------------------------------------- /test/Conversion/tritongpu_to_llvm.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/tritongpu_to_llvm.mlir -------------------------------------------------------------------------------- /test/Conversion/tritongpu_to_llvm_blackwell.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/tritongpu_to_llvm_blackwell.mlir -------------------------------------------------------------------------------- /test/Conversion/tritongpu_to_llvm_debug.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/tritongpu_to_llvm_debug.mlir -------------------------------------------------------------------------------- /test/Conversion/tritongpu_to_llvm_hopper.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/tritongpu_to_llvm_hopper.mlir -------------------------------------------------------------------------------- /test/Conversion/tritongpu_to_llvm_sm120.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/tritongpu_to_llvm_sm120.mlir -------------------------------------------------------------------------------- /test/Conversion/tritongpu_to_llvm_volta.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/tritongpu_to_llvm_volta.mlir -------------------------------------------------------------------------------- /test/Conversion/tritongpu_to_ptx.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/tritongpu_to_ptx.mlir -------------------------------------------------------------------------------- /test/Conversion/tritongpu_to_ptx_mmav3.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/tritongpu_to_ptx_mmav3.mlir -------------------------------------------------------------------------------- /test/Conversion/tritoninstrument_to_llvm.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/tritoninstrument_to_llvm.mlir -------------------------------------------------------------------------------- /test/Conversion/tritonnvidiagpu_to_llvm.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/tritonnvidiagpu_to_llvm.mlir -------------------------------------------------------------------------------- /test/Conversion/ttg_warp_specialize.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/ttg_warp_specialize.mlir -------------------------------------------------------------------------------- /test/Conversion/warp_specialize_to_llvm.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Conversion/warp_specialize_to_llvm.mlir -------------------------------------------------------------------------------- /test/Gluon/auto_encoding.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Gluon/auto_encoding.mlir -------------------------------------------------------------------------------- /test/Gluon/inlining.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Gluon/inlining.mlir -------------------------------------------------------------------------------- /test/Gluon/invalid_auto_encoding.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Gluon/invalid_auto_encoding.mlir -------------------------------------------------------------------------------- /test/Hopper/CMakeLists.txt: -------------------------------------------------------------------------------- 1 | add_subdirectory(WarpSpecialization) 2 | -------------------------------------------------------------------------------- /test/Hopper/WarpSpecialization/1D_tmem.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Hopper/WarpSpecialization/1D_tmem.mlir -------------------------------------------------------------------------------- /test/LLVMIR/break-phi-struct.ll: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/LLVMIR/break-phi-struct.ll -------------------------------------------------------------------------------- /test/NVWS/assign_stage_phase.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/NVWS/assign_stage_phase.mlir -------------------------------------------------------------------------------- /test/NVWS/insert_aref.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/NVWS/insert_aref.mlir -------------------------------------------------------------------------------- /test/NVWS/invalid.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/NVWS/invalid.mlir -------------------------------------------------------------------------------- /test/NVWS/lower_aref.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/NVWS/lower_aref.mlir -------------------------------------------------------------------------------- /test/NVWS/lower_warp_group.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/NVWS/lower_warp_group.mlir -------------------------------------------------------------------------------- /test/NVWS/ops.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/NVWS/ops.mlir -------------------------------------------------------------------------------- /test/Proton/allocate_global_scratch_buffer.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Proton/allocate_global_scratch_buffer.mlir -------------------------------------------------------------------------------- /test/Proton/allocate_shared_memory.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Proton/allocate_shared_memory.mlir -------------------------------------------------------------------------------- /test/Proton/amd/add_sched_barriers.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Proton/amd/add_sched_barriers.mlir -------------------------------------------------------------------------------- /test/Proton/amd/protongpu_to_llvm.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Proton/amd/protongpu_to_llvm.mlir -------------------------------------------------------------------------------- /test/Proton/nvidia/protongpu_to_llvm.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Proton/nvidia/protongpu_to_llvm.mlir -------------------------------------------------------------------------------- /test/Proton/ops.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Proton/ops.mlir -------------------------------------------------------------------------------- /test/Proton/proton_to_protongpu.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Proton/proton_to_protongpu.mlir -------------------------------------------------------------------------------- /test/Proton/protongpu_transforms.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Proton/protongpu_transforms.mlir -------------------------------------------------------------------------------- /test/Proton/scope_id.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Proton/scope_id.mlir -------------------------------------------------------------------------------- /test/TLX/attach-metadata.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TLX/attach-metadata.mlir -------------------------------------------------------------------------------- /test/TLX/insert-require-layout.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TLX/insert-require-layout.mlir -------------------------------------------------------------------------------- /test/TLX/insert_cluster_sync_ops.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TLX/insert_cluster_sync_ops.mlir -------------------------------------------------------------------------------- /test/TLX/ops.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TLX/ops.mlir -------------------------------------------------------------------------------- /test/TLX/propagate-layout.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TLX/propagate-layout.mlir -------------------------------------------------------------------------------- /test/TLX/rewrite-local-alias.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TLX/rewrite-local-alias.mlir -------------------------------------------------------------------------------- /test/TLX/tlx-verifier.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TLX/tlx-verifier.mlir -------------------------------------------------------------------------------- /test/Tools/tensor_layout_print.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Tools/tensor_layout_print.mlir -------------------------------------------------------------------------------- /test/Triton/canonicalize.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Triton/canonicalize.mlir -------------------------------------------------------------------------------- /test/Triton/combine.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Triton/combine.mlir -------------------------------------------------------------------------------- /test/Triton/invalid.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Triton/invalid.mlir -------------------------------------------------------------------------------- /test/Triton/loop-invariant-code-motion.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Triton/loop-invariant-code-motion.mlir -------------------------------------------------------------------------------- /test/Triton/loop-peeling.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Triton/loop-peeling.mlir -------------------------------------------------------------------------------- /test/Triton/loop-unroll.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Triton/loop-unroll.mlir -------------------------------------------------------------------------------- /test/Triton/loop_cse.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Triton/loop_cse.mlir -------------------------------------------------------------------------------- /test/Triton/ops.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Triton/ops.mlir -------------------------------------------------------------------------------- /test/Triton/reorder-broadcast.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Triton/reorder-broadcast.mlir -------------------------------------------------------------------------------- /test/Triton/reproducer.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Triton/reproducer.mlir -------------------------------------------------------------------------------- /test/Triton/rewrite-tensor-pointer.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Triton/rewrite-tensor-pointer.mlir -------------------------------------------------------------------------------- /test/Triton/vecadd.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Triton/vecadd.mlir -------------------------------------------------------------------------------- /test/Triton/verify-make-range.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/Triton/verify-make-range.mlir -------------------------------------------------------------------------------- /test/TritonGPU/accelerate-matmul.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/accelerate-matmul.mlir -------------------------------------------------------------------------------- /test/TritonGPU/accumulator-init.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/accumulator-init.mlir -------------------------------------------------------------------------------- /test/TritonGPU/amd/accelerate-amd-matmul-fma.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/amd/accelerate-amd-matmul-fma.mlir -------------------------------------------------------------------------------- /test/TritonGPU/amd/accelerate-amd-matmul-mfma.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/amd/accelerate-amd-matmul-mfma.mlir -------------------------------------------------------------------------------- /test/TritonGPU/amd/amd-block-pingpong.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/amd/amd-block-pingpong.mlir -------------------------------------------------------------------------------- /test/TritonGPU/amd/amd-canonicalize-pointers.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/amd/amd-canonicalize-pointers.mlir -------------------------------------------------------------------------------- /test/TritonGPU/amd/amd-coalesce-async-copy.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/amd/amd-coalesce-async-copy.mlir -------------------------------------------------------------------------------- /test/TritonGPU/amd/amd-concat-op.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/amd/amd-concat-op.mlir -------------------------------------------------------------------------------- /test/TritonGPU/amd/amd-conditional-barrier.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/amd/amd-conditional-barrier.mlir -------------------------------------------------------------------------------- /test/TritonGPU/amd/amd-convert-buffer-ops.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/amd/amd-convert-buffer-ops.mlir -------------------------------------------------------------------------------- /test/TritonGPU/amd/amd-extractslice-op.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/amd/amd-extractslice-op.mlir -------------------------------------------------------------------------------- /test/TritonGPU/amd/amd-fold-true-cmpi.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/amd/amd-fold-true-cmpi.mlir -------------------------------------------------------------------------------- /test/TritonGPU/amd/amd-hoist-cvtToDotOp.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/amd/amd-hoist-cvtToDotOp.mlir -------------------------------------------------------------------------------- /test/TritonGPU/amd/amd-optimize-dot-operands.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/amd/amd-optimize-dot-operands.mlir -------------------------------------------------------------------------------- /test/TritonGPU/amd/amd-optimize-epilogue.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/amd/amd-optimize-epilogue.mlir -------------------------------------------------------------------------------- /test/TritonGPU/amd/amd-pipeline-chained-dots.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/amd/amd-pipeline-chained-dots.mlir -------------------------------------------------------------------------------- /test/TritonGPU/amd/amd-range-analysis.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/amd/amd-range-analysis.mlir -------------------------------------------------------------------------------- /test/TritonGPU/amd/amd-reorder-instructions.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/amd/amd-reorder-instructions.mlir -------------------------------------------------------------------------------- /test/TritonGPU/amd/amd-schedule-hint.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/amd/amd-schedule-hint.mlir -------------------------------------------------------------------------------- /test/TritonGPU/amd/amd-stream-loop-assume.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/amd/amd-stream-loop-assume.mlir -------------------------------------------------------------------------------- /test/TritonGPU/amd/amd-stream-prefetch.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/amd/amd-stream-prefetch.mlir -------------------------------------------------------------------------------- /test/TritonGPU/amd/in-thread-transpose.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/amd/in-thread-transpose.mlir -------------------------------------------------------------------------------- /test/TritonGPU/amd/invalid.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/amd/invalid.mlir -------------------------------------------------------------------------------- /test/TritonGPU/amd/mfma-double-rate.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/amd/mfma-double-rate.mlir -------------------------------------------------------------------------------- /test/TritonGPU/amd/mfma-xf32.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/amd/mfma-xf32.mlir -------------------------------------------------------------------------------- /test/TritonGPU/amd/optimize-lds-usage.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/amd/optimize-lds-usage.mlir -------------------------------------------------------------------------------- /test/TritonGPU/amd/sink-setprio-mfma.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/amd/sink-setprio-mfma.mlir -------------------------------------------------------------------------------- /test/TritonGPU/atomic-cas.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/atomic-cas.mlir -------------------------------------------------------------------------------- /test/TritonGPU/attention-dp-loop-schedule.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/attention-dp-loop-schedule.mlir -------------------------------------------------------------------------------- /test/TritonGPU/automatic-warp-specialization.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/automatic-warp-specialization.mlir -------------------------------------------------------------------------------- /test/TritonGPU/bf16x3-matmul.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/bf16x3-matmul.mlir -------------------------------------------------------------------------------- /test/TritonGPU/canonicalize.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/canonicalize.mlir -------------------------------------------------------------------------------- /test/TritonGPU/coalesce-async-copy.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/coalesce-async-copy.mlir -------------------------------------------------------------------------------- /test/TritonGPU/coalesce.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/coalesce.mlir -------------------------------------------------------------------------------- /test/TritonGPU/combine-select-if.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/combine-select-if.mlir -------------------------------------------------------------------------------- /test/TritonGPU/combine.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/combine.mlir -------------------------------------------------------------------------------- /test/TritonGPU/consan-negative.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/consan-negative.mlir -------------------------------------------------------------------------------- /test/TritonGPU/consan.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/consan.mlir -------------------------------------------------------------------------------- /test/TritonGPU/dot-operands.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/dot-operands.mlir -------------------------------------------------------------------------------- /test/TritonGPU/fence-inserstion.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/fence-inserstion.mlir -------------------------------------------------------------------------------- /test/TritonGPU/fuse-nested-loops.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/fuse-nested-loops.mlir -------------------------------------------------------------------------------- /test/TritonGPU/global_scratch_alloc.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/global_scratch_alloc.mlir -------------------------------------------------------------------------------- /test/TritonGPU/global_scratch_to_llvm.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/global_scratch_to_llvm.mlir -------------------------------------------------------------------------------- /test/TritonGPU/hoist-tmem-alloc.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/hoist-tmem-alloc.mlir -------------------------------------------------------------------------------- /test/TritonGPU/inline.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/inline.mlir -------------------------------------------------------------------------------- /test/TritonGPU/invalid-attributes.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/invalid-attributes.mlir -------------------------------------------------------------------------------- /test/TritonGPU/invalid.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/invalid.mlir -------------------------------------------------------------------------------- /test/TritonGPU/load-mma-specialization.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/load-mma-specialization.mlir -------------------------------------------------------------------------------- /test/TritonGPU/loop-pipeline-async-latencies.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/loop-pipeline-async-latencies.mlir -------------------------------------------------------------------------------- /test/TritonGPU/loop-pipeline-blackwell.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/loop-pipeline-blackwell.mlir -------------------------------------------------------------------------------- /test/TritonGPU/loop-pipeline-combine-waits.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/loop-pipeline-combine-waits.mlir -------------------------------------------------------------------------------- /test/TritonGPU/loop-pipeline-cuda.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/loop-pipeline-cuda.mlir -------------------------------------------------------------------------------- /test/TritonGPU/loop-pipeline-expand.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/loop-pipeline-expand.mlir -------------------------------------------------------------------------------- /test/TritonGPU/loop-pipeline-hip.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/loop-pipeline-hip.mlir -------------------------------------------------------------------------------- /test/TritonGPU/loop-pipeline-hopper.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/loop-pipeline-hopper.mlir -------------------------------------------------------------------------------- /test/TritonGPU/loop-pipeline-indirect-load.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/loop-pipeline-indirect-load.mlir -------------------------------------------------------------------------------- /test/TritonGPU/loop-pipeline.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/loop-pipeline.mlir -------------------------------------------------------------------------------- /test/TritonGPU/loop-schedule.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/loop-schedule.mlir -------------------------------------------------------------------------------- /test/TritonGPU/matmul-loop-pipeline.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/matmul-loop-pipeline.mlir -------------------------------------------------------------------------------- /test/TritonGPU/matmul.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/matmul.mlir -------------------------------------------------------------------------------- /test/TritonGPU/memdesc-subview-split.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/memdesc-subview-split.mlir -------------------------------------------------------------------------------- /test/TritonGPU/metaws-loop-schedule.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/metaws-loop-schedule.mlir -------------------------------------------------------------------------------- /test/TritonGPU/ops.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/ops.mlir -------------------------------------------------------------------------------- /test/TritonGPU/optimize-locality.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/optimize-locality.mlir -------------------------------------------------------------------------------- /test/TritonGPU/optimize-partition-warps.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/optimize-partition-warps.mlir -------------------------------------------------------------------------------- /test/TritonGPU/optimize_epilogue.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/optimize_epilogue.mlir -------------------------------------------------------------------------------- /test/TritonGPU/partition-loops.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/partition-loops.mlir -------------------------------------------------------------------------------- /test/TritonGPU/partition-scheduling.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/partition-scheduling.mlir -------------------------------------------------------------------------------- /test/TritonGPU/pipeline-assign-latencies.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/pipeline-assign-latencies.mlir -------------------------------------------------------------------------------- /test/TritonGPU/pipeline-loop-nest.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/pipeline-loop-nest.mlir -------------------------------------------------------------------------------- /test/TritonGPU/pipeline-lower-loop.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/pipeline-lower-loop.mlir -------------------------------------------------------------------------------- /test/TritonGPU/pipeline-schedule-loop.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/pipeline-schedule-loop.mlir -------------------------------------------------------------------------------- /test/TritonGPU/prefetch.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/prefetch.mlir -------------------------------------------------------------------------------- /test/TritonGPU/promote-lhs-to-tmem.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/promote-lhs-to-tmem.mlir -------------------------------------------------------------------------------- /test/TritonGPU/prox_fence_insertion.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/prox_fence_insertion.mlir -------------------------------------------------------------------------------- /test/TritonGPU/reduce-data-duplication.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/reduce-data-duplication.mlir -------------------------------------------------------------------------------- /test/TritonGPU/reorder-instructions.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/reorder-instructions.mlir -------------------------------------------------------------------------------- /test/TritonGPU/rewrite-partition-dependencies.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/rewrite-partition-dependencies.mlir -------------------------------------------------------------------------------- /test/TritonGPU/samples/simulated-grouped-gemm.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/samples/simulated-grouped-gemm.mlir -------------------------------------------------------------------------------- /test/TritonGPU/tf32x3-matmul.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/tf32x3-matmul.mlir -------------------------------------------------------------------------------- /test/TritonGPU/verify-blocked-layout.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonGPU/verify-blocked-layout.mlir -------------------------------------------------------------------------------- /test/TritonNvidiaGPU/bf16-atomics.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonNvidiaGPU/bf16-atomics.mlir -------------------------------------------------------------------------------- /test/TritonNvidiaGPU/canonicalize.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonNvidiaGPU/canonicalize.mlir -------------------------------------------------------------------------------- /test/TritonNvidiaGPU/inline.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonNvidiaGPU/inline.mlir -------------------------------------------------------------------------------- /test/TritonNvidiaGPU/interleave_tmem.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonNvidiaGPU/interleave_tmem.mlir -------------------------------------------------------------------------------- /test/TritonNvidiaGPU/invalid.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonNvidiaGPU/invalid.mlir -------------------------------------------------------------------------------- /test/TritonNvidiaGPU/membar.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonNvidiaGPU/membar.mlir -------------------------------------------------------------------------------- /test/TritonNvidiaGPU/mma_lowering.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonNvidiaGPU/mma_lowering.mlir -------------------------------------------------------------------------------- /test/TritonNvidiaGPU/ops.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonNvidiaGPU/ops.mlir -------------------------------------------------------------------------------- /test/TritonNvidiaGPU/tma_lowering.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonNvidiaGPU/tma_lowering.mlir -------------------------------------------------------------------------------- /test/TritonNvidiaGPU/tmem_layouts.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/TritonNvidiaGPU/tmem_layouts.mlir -------------------------------------------------------------------------------- /test/include/Analysis/TestAxisInfo.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/include/Analysis/TestAxisInfo.h -------------------------------------------------------------------------------- /test/lib/Analysis/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/lib/Analysis/CMakeLists.txt -------------------------------------------------------------------------------- /test/lib/Analysis/TestAlias.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/lib/Analysis/TestAlias.cpp -------------------------------------------------------------------------------- /test/lib/Analysis/TestAllocation.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/lib/Analysis/TestAllocation.cpp -------------------------------------------------------------------------------- /test/lib/Analysis/TestAxisInfo.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/lib/Analysis/TestAxisInfo.cpp -------------------------------------------------------------------------------- /test/lib/Analysis/TestMembar.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/lib/Analysis/TestMembar.cpp -------------------------------------------------------------------------------- /test/lib/Analysis/TestPrintNesting.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/lib/Analysis/TestPrintNesting.cpp -------------------------------------------------------------------------------- /test/lib/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/lib/CMakeLists.txt -------------------------------------------------------------------------------- /test/lib/Dialect/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/lib/Dialect/CMakeLists.txt -------------------------------------------------------------------------------- /test/lib/Dialect/TestLoopPeeling.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/lib/Dialect/TestLoopPeeling.cpp -------------------------------------------------------------------------------- /test/lib/Instrumentation/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/lib/Instrumentation/CMakeLists.txt -------------------------------------------------------------------------------- /test/lib/Instrumentation/GPUHello.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/lib/Instrumentation/GPUHello.cpp -------------------------------------------------------------------------------- /test/lib/Proton/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/lib/Proton/CMakeLists.txt -------------------------------------------------------------------------------- /test/lib/Proton/TestScopeIdAllocation.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/lib/Proton/TestScopeIdAllocation.cpp -------------------------------------------------------------------------------- /test/lit.cfg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/lit.cfg.py -------------------------------------------------------------------------------- /test/lit.site.cfg.py.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/test/lit.site.cfg.py.in -------------------------------------------------------------------------------- /third_party/amd/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/CMakeLists.txt -------------------------------------------------------------------------------- /third_party/amd/backend/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/amd/backend/compiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/backend/compiler.py -------------------------------------------------------------------------------- /third_party/amd/backend/driver.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/backend/driver.c -------------------------------------------------------------------------------- /third_party/amd/backend/driver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/backend/driver.py -------------------------------------------------------------------------------- /third_party/amd/backend/include/hip/driver_types.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/backend/include/hip/driver_types.h -------------------------------------------------------------------------------- /third_party/amd/backend/include/hip/hip_bf16.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/backend/include/hip/hip_bf16.h -------------------------------------------------------------------------------- /third_party/amd/backend/include/hip/hip_bfloat16.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/backend/include/hip/hip_bfloat16.h -------------------------------------------------------------------------------- /third_party/amd/backend/include/hip/hip_common.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/backend/include/hip/hip_common.h -------------------------------------------------------------------------------- /third_party/amd/backend/include/hip/hip_complex.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/backend/include/hip/hip_complex.h -------------------------------------------------------------------------------- /third_party/amd/backend/include/hip/hip_ext.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/backend/include/hip/hip_ext.h -------------------------------------------------------------------------------- /third_party/amd/backend/include/hip/hip_fp16.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/backend/include/hip/hip_fp16.h -------------------------------------------------------------------------------- /third_party/amd/backend/include/hip/hip_fp8.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/backend/include/hip/hip_fp8.h -------------------------------------------------------------------------------- /third_party/amd/backend/include/hip/hip_hcc.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/backend/include/hip/hip_hcc.h -------------------------------------------------------------------------------- /third_party/amd/backend/include/hip/hip_profile.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/backend/include/hip/hip_profile.h -------------------------------------------------------------------------------- /third_party/amd/backend/include/hip/hip_runtime.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/backend/include/hip/hip_runtime.h -------------------------------------------------------------------------------- /third_party/amd/backend/include/hip/hip_version.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/backend/include/hip/hip_version.h -------------------------------------------------------------------------------- /third_party/amd/backend/include/hip/hiprtc.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/backend/include/hip/hiprtc.h -------------------------------------------------------------------------------- /third_party/amd/backend/include/hsa/Brig.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/backend/include/hsa/Brig.h -------------------------------------------------------------------------------- /third_party/amd/backend/include/hsa/amd_hsa_elf.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/backend/include/hsa/amd_hsa_elf.h -------------------------------------------------------------------------------- /third_party/amd/backend/include/hsa/hsa.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/backend/include/hsa/hsa.h -------------------------------------------------------------------------------- /third_party/amd/backend/include/hsa/hsa_amd_tool.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/backend/include/hsa/hsa_amd_tool.h -------------------------------------------------------------------------------- /third_party/amd/backend/include/hsa/hsa_ext_amd.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/backend/include/hsa/hsa_ext_amd.h -------------------------------------------------------------------------------- /third_party/amd/backend/include/roctracer/roctx.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/backend/include/roctracer/roctx.h -------------------------------------------------------------------------------- /third_party/amd/backend/lib/asanrtl.bc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/backend/lib/asanrtl.bc -------------------------------------------------------------------------------- /third_party/amd/backend/lib/ockl.bc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/backend/lib/ockl.bc -------------------------------------------------------------------------------- /third_party/amd/backend/lib/ocml.bc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/backend/lib/ocml.bc -------------------------------------------------------------------------------- /third_party/amd/include/Analysis/AxisInfoExt.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/include/Analysis/AxisInfoExt.h -------------------------------------------------------------------------------- /third_party/amd/include/Analysis/RangeAnalysis.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/include/Analysis/RangeAnalysis.h -------------------------------------------------------------------------------- /third_party/amd/include/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/include/CMakeLists.txt -------------------------------------------------------------------------------- /third_party/amd/include/Dialect/CMakeLists.txt: -------------------------------------------------------------------------------- 1 | add_subdirectory(TritonAMDGPU) 2 | -------------------------------------------------------------------------------- /third_party/amd/include/Dialect/TritonAMDGPU/CMakeLists.txt: -------------------------------------------------------------------------------- 1 | add_subdirectory(IR) 2 | -------------------------------------------------------------------------------- /third_party/amd/include/Utils/Utility.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/include/Utils/Utility.h -------------------------------------------------------------------------------- /third_party/amd/language/hip/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/language/hip/__init__.py -------------------------------------------------------------------------------- /third_party/amd/language/hip/libdevice.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/language/hip/libdevice.py -------------------------------------------------------------------------------- /third_party/amd/language/hip/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/language/hip/utils.py -------------------------------------------------------------------------------- /third_party/amd/lib/Analysis/AMDGPUAllocation.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/lib/Analysis/AMDGPUAllocation.cpp -------------------------------------------------------------------------------- /third_party/amd/lib/Analysis/AxisInfoExt.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/lib/Analysis/AxisInfoExt.cpp -------------------------------------------------------------------------------- /third_party/amd/lib/Analysis/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/lib/Analysis/CMakeLists.txt -------------------------------------------------------------------------------- /third_party/amd/lib/Analysis/RangeAnalysis.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/lib/Analysis/RangeAnalysis.cpp -------------------------------------------------------------------------------- /third_party/amd/lib/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/lib/CMakeLists.txt -------------------------------------------------------------------------------- /third_party/amd/lib/Dialect/CMakeLists.txt: -------------------------------------------------------------------------------- 1 | add_subdirectory(TritonAMDGPU) 2 | -------------------------------------------------------------------------------- /third_party/amd/lib/TritonAMDGPUToLLVM/Utility.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/lib/TritonAMDGPUToLLVM/Utility.cpp -------------------------------------------------------------------------------- /third_party/amd/lib/TritonAMDGPUToLLVM/Utility.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/lib/TritonAMDGPUToLLVM/Utility.h -------------------------------------------------------------------------------- /third_party/amd/python/test/attn_fwd.ttir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/python/test/attn_fwd.ttir -------------------------------------------------------------------------------- /third_party/amd/python/test/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/python/test/conftest.py -------------------------------------------------------------------------------- /third_party/amd/python/triton_amd.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/python/triton_amd.cc -------------------------------------------------------------------------------- /third_party/amd/test/CMakeLists.txt: -------------------------------------------------------------------------------- 1 | add_subdirectory(lib) 2 | -------------------------------------------------------------------------------- /third_party/amd/test/lib/Analysis/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/test/lib/Analysis/CMakeLists.txt -------------------------------------------------------------------------------- /third_party/amd/test/lib/Analysis/TestAxisInfo.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/test/lib/Analysis/TestAxisInfo.cpp -------------------------------------------------------------------------------- /third_party/amd/test/lib/CMakeLists.txt: -------------------------------------------------------------------------------- 1 | add_subdirectory(Analysis) 2 | -------------------------------------------------------------------------------- /third_party/amd/tools/hip/compile.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/tools/hip/compile.cpp -------------------------------------------------------------------------------- /third_party/amd/tools/hip/compile.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/tools/hip/compile.h -------------------------------------------------------------------------------- /third_party/amd/unittest/CMakeLists.txt: -------------------------------------------------------------------------------- 1 | add_subdirectory(Conversion) 2 | -------------------------------------------------------------------------------- /third_party/amd/unittest/Conversion/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/amd/unittest/Conversion/CMakeLists.txt -------------------------------------------------------------------------------- /third_party/f2reduce/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/f2reduce/CMakeLists.txt -------------------------------------------------------------------------------- /third_party/f2reduce/LICENCE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/f2reduce/LICENCE.txt -------------------------------------------------------------------------------- /third_party/f2reduce/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/f2reduce/README.md -------------------------------------------------------------------------------- /third_party/f2reduce/VERSION: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/f2reduce/VERSION -------------------------------------------------------------------------------- /third_party/f2reduce/f2reduce.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/f2reduce/f2reduce.cpp -------------------------------------------------------------------------------- /third_party/f2reduce/f2reduce.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/f2reduce/f2reduce.h -------------------------------------------------------------------------------- /third_party/nvidia/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/nvidia/CMakeLists.txt -------------------------------------------------------------------------------- /third_party/nvidia/backend/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/nvidia/backend/bin/ptxas-blackwell: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/nvidia/backend/bin/ptxas-blackwell -------------------------------------------------------------------------------- /third_party/nvidia/backend/compiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/nvidia/backend/compiler.py -------------------------------------------------------------------------------- /third_party/nvidia/backend/driver.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/nvidia/backend/driver.c -------------------------------------------------------------------------------- /third_party/nvidia/backend/driver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/nvidia/backend/driver.py -------------------------------------------------------------------------------- /third_party/nvidia/backend/lib/libdevice.10.bc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/nvidia/backend/lib/libdevice.10.bc -------------------------------------------------------------------------------- /third_party/nvidia/hopper/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/nvidia/hopper/CMakeLists.txt -------------------------------------------------------------------------------- /third_party/nvidia/hopper/include/CMakeLists.txt: -------------------------------------------------------------------------------- 1 | add_subdirectory(Transforms) 2 | -------------------------------------------------------------------------------- /third_party/nvidia/hopper/lib/CMakeLists.txt: -------------------------------------------------------------------------------- 1 | add_subdirectory(Transforms) 2 | -------------------------------------------------------------------------------- /third_party/nvidia/include/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/nvidia/include/CMakeLists.txt -------------------------------------------------------------------------------- /third_party/nvidia/include/Dialect/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/nvidia/include/Dialect/CMakeLists.txt -------------------------------------------------------------------------------- /third_party/nvidia/include/Dialect/NVGPU/CMakeLists.txt: -------------------------------------------------------------------------------- 1 | add_subdirectory(IR) 2 | -------------------------------------------------------------------------------- /third_party/nvidia/include/NVGPUToLLVM/Passes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/nvidia/include/NVGPUToLLVM/Passes.h -------------------------------------------------------------------------------- /third_party/nvidia/include/NVGPUToLLVM/Passes.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/nvidia/include/NVGPUToLLVM/Passes.td -------------------------------------------------------------------------------- /third_party/nvidia/include/cublas_instance.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/nvidia/include/cublas_instance.h -------------------------------------------------------------------------------- /third_party/nvidia/include/cublas_types.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/nvidia/include/cublas_types.h -------------------------------------------------------------------------------- /third_party/nvidia/language/cuda/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/nvidia/language/cuda/__init__.py -------------------------------------------------------------------------------- /third_party/nvidia/language/cuda/gdc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/nvidia/language/cuda/gdc.py -------------------------------------------------------------------------------- /third_party/nvidia/language/cuda/libdevice.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/nvidia/language/cuda/libdevice.py -------------------------------------------------------------------------------- /third_party/nvidia/language/cuda/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/nvidia/language/cuda/utils.py -------------------------------------------------------------------------------- /third_party/nvidia/lib/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/nvidia/lib/CMakeLists.txt -------------------------------------------------------------------------------- /third_party/nvidia/lib/Dialect/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/nvidia/lib/Dialect/CMakeLists.txt -------------------------------------------------------------------------------- /third_party/nvidia/lib/Dialect/NVGPU/CMakeLists.txt: -------------------------------------------------------------------------------- 1 | add_subdirectory(IR) 2 | -------------------------------------------------------------------------------- /third_party/nvidia/lib/Dialect/NVWS/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/nvidia/lib/Dialect/NVWS/CMakeLists.txt -------------------------------------------------------------------------------- /third_party/nvidia/lib/Dialect/NVWS/IR/Dialect.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/nvidia/lib/Dialect/NVWS/IR/Dialect.cpp -------------------------------------------------------------------------------- /third_party/nvidia/lib/Dialect/NVWS/IR/Ops.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/nvidia/lib/Dialect/NVWS/IR/Ops.cpp -------------------------------------------------------------------------------- /third_party/nvidia/lib/NVGPUToLLVM/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/nvidia/lib/NVGPUToLLVM/CMakeLists.txt -------------------------------------------------------------------------------- /third_party/nvidia/tools/cuda/compile.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/nvidia/tools/cuda/compile.c -------------------------------------------------------------------------------- /third_party/nvidia/tools/cuda/compile.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/nvidia/tools/cuda/compile.h -------------------------------------------------------------------------------- /third_party/nvidia/triton_nvidia.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/nvidia/triton_nvidia.cc -------------------------------------------------------------------------------- /third_party/nvidia/unittest/CMakeLists.txt: -------------------------------------------------------------------------------- 1 | add_subdirectory(Conversion) 2 | -------------------------------------------------------------------------------- /third_party/nvidia/unittest/Conversion/CMakeLists.txt: -------------------------------------------------------------------------------- 1 | add_subdirectory(TritonGPUToLLVM) 2 | -------------------------------------------------------------------------------- /third_party/proton/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/.gitignore -------------------------------------------------------------------------------- /third_party/proton/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/CMakeLists.txt -------------------------------------------------------------------------------- /third_party/proton/Dialect/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/Dialect/CMakeLists.txt -------------------------------------------------------------------------------- /third_party/proton/Dialect/include/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/Dialect/include/CMakeLists.txt -------------------------------------------------------------------------------- /third_party/proton/Dialect/include/Dialect/Proton/CMakeLists.txt: -------------------------------------------------------------------------------- 1 | add_subdirectory(IR) 2 | -------------------------------------------------------------------------------- /third_party/proton/Dialect/lib/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/Dialect/lib/CMakeLists.txt -------------------------------------------------------------------------------- /third_party/proton/Dialect/lib/Dialect/Proton/CMakeLists.txt: -------------------------------------------------------------------------------- 1 | add_subdirectory(IR) 2 | -------------------------------------------------------------------------------- /third_party/proton/Dialect/triton_proton.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/Dialect/triton_proton.cc -------------------------------------------------------------------------------- /third_party/proton/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/README.md -------------------------------------------------------------------------------- /third_party/proton/common/CMakeLists.txt: -------------------------------------------------------------------------------- 1 | add_subdirectory(lib) 2 | -------------------------------------------------------------------------------- /third_party/proton/common/include/Device.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/common/include/Device.h -------------------------------------------------------------------------------- /third_party/proton/common/lib/CMakeLists.txt: -------------------------------------------------------------------------------- 1 | add_subdirectory(TraceDataIO) 2 | -------------------------------------------------------------------------------- /third_party/proton/csrc/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/csrc/CMakeLists.txt -------------------------------------------------------------------------------- /third_party/proton/csrc/Proton.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/csrc/Proton.cpp -------------------------------------------------------------------------------- /third_party/proton/csrc/include/Context/Context.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/csrc/include/Context/Context.h -------------------------------------------------------------------------------- /third_party/proton/csrc/include/Context/Python.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/csrc/include/Context/Python.h -------------------------------------------------------------------------------- /third_party/proton/csrc/include/Context/Shadow.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/csrc/include/Context/Shadow.h -------------------------------------------------------------------------------- /third_party/proton/csrc/include/Data/Data.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/csrc/include/Data/Data.h -------------------------------------------------------------------------------- /third_party/proton/csrc/include/Data/Metric.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/csrc/include/Data/Metric.h -------------------------------------------------------------------------------- /third_party/proton/csrc/include/Data/TraceData.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/csrc/include/Data/TraceData.h -------------------------------------------------------------------------------- /third_party/proton/csrc/include/Data/TreeData.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/csrc/include/Data/TreeData.h -------------------------------------------------------------------------------- /third_party/proton/csrc/include/Driver/Dispatch.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/csrc/include/Driver/Dispatch.h -------------------------------------------------------------------------------- /third_party/proton/csrc/include/Proton.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/csrc/include/Proton.h -------------------------------------------------------------------------------- /third_party/proton/csrc/include/Session/Session.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/csrc/include/Session/Session.h -------------------------------------------------------------------------------- /third_party/proton/csrc/include/Utility/Atomic.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/csrc/include/Utility/Atomic.h -------------------------------------------------------------------------------- /third_party/proton/csrc/include/Utility/Errors.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/csrc/include/Utility/Errors.h -------------------------------------------------------------------------------- /third_party/proton/csrc/include/Utility/Map.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/csrc/include/Utility/Map.h -------------------------------------------------------------------------------- /third_party/proton/csrc/include/Utility/Numeric.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/csrc/include/Utility/Numeric.h -------------------------------------------------------------------------------- /third_party/proton/csrc/include/Utility/Set.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/csrc/include/Utility/Set.h -------------------------------------------------------------------------------- /third_party/proton/csrc/include/Utility/String.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/csrc/include/Utility/String.h -------------------------------------------------------------------------------- /third_party/proton/csrc/include/Utility/Traits.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/csrc/include/Utility/Traits.h -------------------------------------------------------------------------------- /third_party/proton/csrc/lib/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/csrc/lib/CMakeLists.txt -------------------------------------------------------------------------------- /third_party/proton/csrc/lib/Context/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/csrc/lib/Context/CMakeLists.txt -------------------------------------------------------------------------------- /third_party/proton/csrc/lib/Context/Context.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/csrc/lib/Context/Context.cpp -------------------------------------------------------------------------------- /third_party/proton/csrc/lib/Context/Python.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/csrc/lib/Context/Python.cpp -------------------------------------------------------------------------------- /third_party/proton/csrc/lib/Context/Shadow.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/csrc/lib/Context/Shadow.cpp -------------------------------------------------------------------------------- /third_party/proton/csrc/lib/Data/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/csrc/lib/Data/CMakeLists.txt -------------------------------------------------------------------------------- /third_party/proton/csrc/lib/Data/Data.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/csrc/lib/Data/Data.cpp -------------------------------------------------------------------------------- /third_party/proton/csrc/lib/Data/TraceData.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/csrc/lib/Data/TraceData.cpp -------------------------------------------------------------------------------- /third_party/proton/csrc/lib/Data/TreeData.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/csrc/lib/Data/TreeData.cpp -------------------------------------------------------------------------------- /third_party/proton/csrc/lib/Driver/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/csrc/lib/Driver/CMakeLists.txt -------------------------------------------------------------------------------- /third_party/proton/csrc/lib/Driver/Device.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/csrc/lib/Driver/Device.cpp -------------------------------------------------------------------------------- /third_party/proton/csrc/lib/Driver/GPU/CudaApi.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/csrc/lib/Driver/GPU/CudaApi.cpp -------------------------------------------------------------------------------- /third_party/proton/csrc/lib/Driver/GPU/HipApi.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/csrc/lib/Driver/GPU/HipApi.cpp -------------------------------------------------------------------------------- /third_party/proton/csrc/lib/Driver/GPU/HsaApi.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/csrc/lib/Driver/GPU/HsaApi.cpp -------------------------------------------------------------------------------- /third_party/proton/csrc/lib/Session/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/csrc/lib/Session/CMakeLists.txt -------------------------------------------------------------------------------- /third_party/proton/csrc/lib/Session/Session.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/csrc/lib/Session/Session.cpp -------------------------------------------------------------------------------- /third_party/proton/proton/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/proton/__init__.py -------------------------------------------------------------------------------- /third_party/proton/proton/context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/proton/context.py -------------------------------------------------------------------------------- /third_party/proton/proton/flags.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/proton/flags.py -------------------------------------------------------------------------------- /third_party/proton/proton/hooks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/proton/hooks/__init__.py -------------------------------------------------------------------------------- /third_party/proton/proton/hooks/hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/proton/hooks/hook.py -------------------------------------------------------------------------------- /third_party/proton/proton/hooks/instrumentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/proton/hooks/instrumentation.py -------------------------------------------------------------------------------- /third_party/proton/proton/hooks/launch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/proton/hooks/launch.py -------------------------------------------------------------------------------- /third_party/proton/proton/language.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/proton/language.py -------------------------------------------------------------------------------- /third_party/proton/proton/mode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/proton/mode.py -------------------------------------------------------------------------------- /third_party/proton/proton/profile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/proton/profile.py -------------------------------------------------------------------------------- /third_party/proton/proton/proton.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/proton/proton.py -------------------------------------------------------------------------------- /third_party/proton/proton/scope.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/proton/scope.py -------------------------------------------------------------------------------- /third_party/proton/proton/specs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/proton/specs.py -------------------------------------------------------------------------------- /third_party/proton/proton/state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/proton/state.py -------------------------------------------------------------------------------- /third_party/proton/proton/viewer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/proton/viewer.py -------------------------------------------------------------------------------- /third_party/proton/scripts/dump_ttgir.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/scripts/dump_ttgir.sh -------------------------------------------------------------------------------- /third_party/proton/test/CMakeLists.txt: -------------------------------------------------------------------------------- 1 | add_subdirectory(unittest) 2 | -------------------------------------------------------------------------------- /third_party/proton/test/examples/cuda.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/test/examples/cuda.json -------------------------------------------------------------------------------- /third_party/proton/test/examples/frame.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/test/examples/frame.json -------------------------------------------------------------------------------- /third_party/proton/test/examples/hip.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/test/examples/hip.json -------------------------------------------------------------------------------- /third_party/proton/test/examples/leaf_nodes.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/test/examples/leaf_nodes.json -------------------------------------------------------------------------------- /third_party/proton/test/examples/triton.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/test/examples/triton.json -------------------------------------------------------------------------------- /third_party/proton/test/helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/test/helper.py -------------------------------------------------------------------------------- /third_party/proton/test/helper_kernels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/test/helper_kernels.py -------------------------------------------------------------------------------- /third_party/proton/test/override_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/test/override_helper.py -------------------------------------------------------------------------------- /third_party/proton/test/test_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/test/test_api.py -------------------------------------------------------------------------------- /third_party/proton/test/test_cmd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/test/test_cmd.py -------------------------------------------------------------------------------- /third_party/proton/test/test_instrumentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/test/test_instrumentation.py -------------------------------------------------------------------------------- /third_party/proton/test/test_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/test/test_lib.py -------------------------------------------------------------------------------- /third_party/proton/test/test_override.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/test/test_override.py -------------------------------------------------------------------------------- /third_party/proton/test/test_profile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/test/test_profile.py -------------------------------------------------------------------------------- /third_party/proton/test/test_viewer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/test/test_viewer.py -------------------------------------------------------------------------------- /third_party/proton/test/unittest/CMakeLists.txt: -------------------------------------------------------------------------------- 1 | add_subdirectory(TraceDataIO) 2 | -------------------------------------------------------------------------------- /third_party/proton/test/unittest/util/loop.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/test/unittest/util/loop.bin -------------------------------------------------------------------------------- /third_party/proton/test/unittest/util/seq.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/test/unittest/util/seq.bin -------------------------------------------------------------------------------- /third_party/proton/test/unittest/util/trace_gen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/test/unittest/util/trace_gen.py -------------------------------------------------------------------------------- /third_party/proton/tutorials/dynamic-net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/tutorials/dynamic-net.py -------------------------------------------------------------------------------- /third_party/proton/tutorials/matmul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/proton/tutorials/matmul.py -------------------------------------------------------------------------------- /third_party/tlx/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/tlx/CMakeLists.txt -------------------------------------------------------------------------------- /third_party/tlx/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/tlx/README.md -------------------------------------------------------------------------------- /third_party/tlx/denoise.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/tlx/denoise.sh -------------------------------------------------------------------------------- /third_party/tlx/dialect/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/tlx/dialect/CMakeLists.txt -------------------------------------------------------------------------------- /third_party/tlx/dialect/include/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/tlx/dialect/include/CMakeLists.txt -------------------------------------------------------------------------------- /third_party/tlx/dialect/include/IR/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/tlx/dialect/include/IR/CMakeLists.txt -------------------------------------------------------------------------------- /third_party/tlx/dialect/include/IR/Dialect.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/tlx/dialect/include/IR/Dialect.h -------------------------------------------------------------------------------- /third_party/tlx/dialect/include/IR/TLXAttrDefs.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/tlx/dialect/include/IR/TLXAttrDefs.td -------------------------------------------------------------------------------- /third_party/tlx/dialect/include/IR/TLXOps.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/tlx/dialect/include/IR/TLXOps.td -------------------------------------------------------------------------------- /third_party/tlx/dialect/include/IR/Traits.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/tlx/dialect/include/IR/Traits.h -------------------------------------------------------------------------------- /third_party/tlx/dialect/lib/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/tlx/dialect/lib/CMakeLists.txt -------------------------------------------------------------------------------- /third_party/tlx/dialect/lib/IR/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/tlx/dialect/lib/IR/CMakeLists.txt -------------------------------------------------------------------------------- /third_party/tlx/dialect/lib/IR/Dialect.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/tlx/dialect/lib/IR/Dialect.cpp -------------------------------------------------------------------------------- /third_party/tlx/dialect/lib/IR/Ops.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/tlx/dialect/lib/IR/Ops.cpp -------------------------------------------------------------------------------- /third_party/tlx/dialect/lib/IR/Traits.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/tlx/dialect/lib/IR/Traits.cpp -------------------------------------------------------------------------------- /third_party/tlx/dialect/triton_tlx.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/tlx/dialect/triton_tlx.cc -------------------------------------------------------------------------------- /third_party/tlx/doc/TLX-triton-conference.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/tlx/doc/TLX-triton-conference.pdf -------------------------------------------------------------------------------- /third_party/tlx/doc/tlx_barriers.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/tlx/doc/tlx_barriers.md -------------------------------------------------------------------------------- /third_party/tlx/language/tlx/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/tlx/language/tlx/__init__.py -------------------------------------------------------------------------------- /third_party/tlx/language/tlx/barrier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/tlx/language/tlx/barrier.py -------------------------------------------------------------------------------- /third_party/tlx/language/tlx/dynamic_launch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/tlx/language/tlx/dynamic_launch.py -------------------------------------------------------------------------------- /third_party/tlx/language/tlx/mem_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/tlx/language/tlx/mem_ops.py -------------------------------------------------------------------------------- /third_party/tlx/language/tlx/mma_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/tlx/language/tlx/mma_ops.py -------------------------------------------------------------------------------- /third_party/tlx/language/tlx/types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/tlx/language/tlx/types.py -------------------------------------------------------------------------------- /third_party/tlx/language/tlx/utility.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/tlx/language/tlx/utility.py -------------------------------------------------------------------------------- /third_party/tlx/media/image1.PNG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/tlx/media/image1.PNG -------------------------------------------------------------------------------- /third_party/tlx/media/image2.PNG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/tlx/media/image2.PNG -------------------------------------------------------------------------------- /third_party/tlx/media/image3.PNG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/tlx/media/image3.PNG -------------------------------------------------------------------------------- /third_party/tlx/media/image4.PNG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/tlx/media/image4.PNG -------------------------------------------------------------------------------- /third_party/tlx/media/image5.PNG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/tlx/media/image5.PNG -------------------------------------------------------------------------------- /third_party/tlx/run_all.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/tlx/run_all.sh -------------------------------------------------------------------------------- /third_party/tlx/tutorials/.gitignore: -------------------------------------------------------------------------------- 1 | *.chrome_trace 2 | -------------------------------------------------------------------------------- /third_party/tlx/tutorials/amd-gemm-pipelined.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/tlx/tutorials/amd-gemm-pipelined.py -------------------------------------------------------------------------------- /third_party/tlx/tutorials/blackwell-gdpa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/tlx/tutorials/blackwell-gdpa.py -------------------------------------------------------------------------------- /third_party/tlx/tutorials/hopper-fa-ws_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/tlx/tutorials/hopper-fa-ws_test.py -------------------------------------------------------------------------------- /third_party/tlx/tutorials/vector-add2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/third_party/tlx/tutorials/vector-add2.py -------------------------------------------------------------------------------- /unittest/Analysis/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/unittest/Analysis/CMakeLists.txt -------------------------------------------------------------------------------- /unittest/Analysis/UtilityTest.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/unittest/Analysis/UtilityTest.cpp -------------------------------------------------------------------------------- /unittest/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/unittest/CMakeLists.txt -------------------------------------------------------------------------------- /unittest/Dialect/CMakeLists.txt: -------------------------------------------------------------------------------- 1 | add_subdirectory(TritonGPU) 2 | -------------------------------------------------------------------------------- /unittest/Dialect/TritonGPU/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/unittest/Dialect/TritonGPU/CMakeLists.txt -------------------------------------------------------------------------------- /unittest/Dialect/TritonGPU/DialectTest.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/unittest/Dialect/TritonGPU/DialectTest.cpp -------------------------------------------------------------------------------- /unittest/Dialect/TritonGPU/DumpLayoutTest.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/unittest/Dialect/TritonGPU/DumpLayoutTest.cpp -------------------------------------------------------------------------------- /unittest/Dialect/TritonGPU/SwizzleTest.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/unittest/Dialect/TritonGPU/SwizzleTest.cpp -------------------------------------------------------------------------------- /unittest/Tools/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/unittest/Tools/CMakeLists.txt -------------------------------------------------------------------------------- /unittest/Tools/LayoutUtilsTest.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/unittest/Tools/LayoutUtilsTest.cpp -------------------------------------------------------------------------------- /unittest/Tools/LinearLayoutTest.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/unittest/Tools/LinearLayoutTest.cpp -------------------------------------------------------------------------------- /unittest/googletest.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/unittest/googletest.cmake -------------------------------------------------------------------------------- /utils/generate-test-checks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/utils/generate-test-checks.py -------------------------------------------------------------------------------- /utils/nightly.pypirc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookexperimental/triton/HEAD/utils/nightly.pypirc --------------------------------------------------------------------------------