saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[Vulkan] Add -Os flag to shader compilation (#57199)

Browse source 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/57199

Reduces the size of compiled shaders, and also potentially adds some performance boost.

Test Plan: Imported from OSS

Reviewed By: xta0

Differential Revision: D28293816

Pulled By: SS-JIA

fbshipit-source-id: 424dc0bce24d6115ba2bf8405027e967f6cb9497
This commit is contained in:
Sicheng Stephen Jia 2021-05-07 13:13:11 -07:00 committed by Facebook GitHub Bot
parent 731dcd75f5
commit 96fce78ac4
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
aten/src/ATen/gen_vulkan_spv.py
View file

@ -43,7 +43,7 @@ def genCppH(hFilePath, cppFilePath, srcDirPath, glslcPath, tmpDirPath, env):
print("spvPath {}".format(spvPath))
cmd = [
glslcPath, "-fshader-stage=compute",
glslcPath, "-fshader-stage=compute", "-Os",
srcPath, "-o", spvPath,
"--target-env=vulkan1.0",
"-Werror"