Skip to content

Commit 4278a34

Browse files
atalmanfacebook-github-bot
authored andcommitted
Revert D69257102: Unify cuBLASLt workspaces with cuBLAS workspaces (#145130)
Differential Revision: D69257102 Original commit changeset: 4a2e6391fa89 Original Phabricator Diff: D69257102 fbshipit-source-id: 6fb20e28aafa38376b586e55782bdac72c45f890
1 parent dab71c0 commit 4278a34

File tree

1 file changed

+0
-9
lines changed

1 file changed

+0
-9
lines changed

userbenchmark/dynamo/dynamobench/common.py

Lines changed: 0 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -3563,15 +3563,6 @@ def run(runner, args, original_dir=None):
35633563
# some of the models do not support use_deterministic_algorithms
35643564
torch.use_deterministic_algorithms(True)
35653565
os.environ["CUBLAS_WORKSPACE_CONFIG"] = ":4096:8"
3566-
if args.only is not None and args.only in {
3567-
"DebertaForQuestionAnswering",
3568-
"RobertaForQuestionAnswering",
3569-
"nvidia_deeprecommender",
3570-
"volo_d1_224",
3571-
}:
3572-
# These seem unhappy with numerics of larger cuBLASLt workspace
3573-
# sizes following #145130 (due to enabling split-k?)
3574-
torch.backends.cuda.matmul.allow_fp16_reduced_precision_reduction = False
35753566
torch.backends.cudnn.deterministic = True
35763567
torch.backends.cudnn.allow_tf32 = False
35773568
torch.backends.cudnn.benchmark = False

0 commit comments

Comments
 (0)