Unverified Commit 97a09c29 authored by Teddy Do's avatar Teddy Do Committed by GitHub
Browse files

Fix ptxas compilation on sm103 for triton kernels (#2539)



* add triton ptxas path  for gb300 to find where it is to avoid compilation errors
Signed-off-by: default avatartdophung <tdophung@nvidia.com>

* add these flags in advance to preven future breaks when ops are extended to multi gpus
Signed-off-by: default avatartdophung <tdophung@nvidia.com>

* add this also to L1
Signed-off-by: default avatartdophung <tdophung@nvidia.com>

---------
Signed-off-by: default avatartdophung <tdophung@nvidia.com>
parent eb8e792b
# Copyright (c) 2022-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved. # Copyright (c) 2022-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# #
# See LICENSE for license information. # See LICENSE for license information.
export TRITON_PTXAS_PATH=/usr/local/cuda/bin/ptxas
function error_exit() { function error_exit() {
echo "Error: $1" echo "Error: $1"
......
# Copyright (c) 2022-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved. # Copyright (c) 2022-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# #
# See LICENSE for license information. # See LICENSE for license information.
export TRITON_PTXAS_PATH=/usr/local/cuda/bin/ptxas
set -x set -x
......
# Copyright (c) 2022-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved. # Copyright (c) 2022-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# #
# See LICENSE for license information. # See LICENSE for license information.
export TRITON_PTXAS_PATH=/usr/local/cuda/bin/ptxas
function test_fail() { function test_fail() {
RET=1 RET=1
......
# Copyright (c) 2022-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved. # Copyright (c) 2022-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# #
# See LICENSE for license information. # See LICENSE for license information.
export TRITON_PTXAS_PATH=/usr/local/cuda/bin/ptxas
set -xe set -xe
......
# Copyright (c) 2022-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved. # Copyright (c) 2022-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# #
# See LICENSE for license information. # See LICENSE for license information.
export TRITON_PTXAS_PATH=/usr/local/cuda/bin/ptxas
set -x set -x
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment