Skip to content

Commit

Permalink
AMD changed the default pipeliner behavior in OSS Triton so num_stage…
Browse files Browse the repository at this point in the history
…s=0 triggers an assert.

Reviewed By: bertmaher

Differential Revision: D65695544

fbshipit-source-id: 95988b90debba743f90bfca3a63fa9f34df5e492
  • Loading branch information
SamGinzburg authored and facebook-github-bot committed Nov 12, 2024
1 parent 359dfb4 commit 3c83e0b
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 10 deletions.
10 changes: 5 additions & 5 deletions tritonbench/operators/bf16xint16_gemm/kernel.py
Original file line number Diff line number Diff line change
Expand Up @@ -194,7 +194,7 @@ def get_hip_autotune_config():
"waves_per_eu": 2,
},
num_warps=4,
num_stages=0,
num_stages=2,
),
triton.Config(
{
Expand All @@ -205,7 +205,7 @@ def get_hip_autotune_config():
"waves_per_eu": 2,
},
num_warps=8,
num_stages=0,
num_stages=2,
),
triton.Config(
{
Expand All @@ -216,7 +216,7 @@ def get_hip_autotune_config():
"waves_per_eu": 2,
},
num_warps=8,
num_stages=0,
num_stages=2,
),
triton.Config(
{
Expand All @@ -227,7 +227,7 @@ def get_hip_autotune_config():
"waves_per_eu": 3,
},
num_warps=4,
num_stages=0,
num_stages=2,
),
triton.Config(
{
Expand All @@ -238,7 +238,7 @@ def get_hip_autotune_config():
"waves_per_eu": 8,
},
num_warps=4,
num_stages=0,
num_stages=2,
),
]

Expand Down
10 changes: 5 additions & 5 deletions tritonbench/operators/fp8_gemm/tutorial.py
Original file line number Diff line number Diff line change
Expand Up @@ -341,7 +341,7 @@ def get_hip_autotune_config():
"waves_per_eu": 2,
},
num_warps=4,
num_stages=0,
num_stages=2,
),
triton.Config(
{
Expand All @@ -352,7 +352,7 @@ def get_hip_autotune_config():
"waves_per_eu": 2,
},
num_warps=8,
num_stages=0,
num_stages=2,
),
triton.Config(
{
Expand All @@ -363,7 +363,7 @@ def get_hip_autotune_config():
"waves_per_eu": 2,
},
num_warps=8,
num_stages=0,
num_stages=2,
),
triton.Config(
{
Expand All @@ -374,7 +374,7 @@ def get_hip_autotune_config():
"waves_per_eu": 3,
},
num_warps=4,
num_stages=0,
num_stages=2,
),
triton.Config(
{
Expand All @@ -385,7 +385,7 @@ def get_hip_autotune_config():
"waves_per_eu": 8,
},
num_warps=4,
num_stages=0,
num_stages=2,
),
]

Expand Down

0 comments on commit 3c83e0b

Please sign in to comment.