Commit450fc8f

authored

build-te.sh: stop injecting -real (#1746)

SinceNVIDIA/TransformerEngine#2279 this is nolonger compatible with TransformerEngine's build system.

1 parentd39fa23 commit450fc8fCopy full SHA for 450fc8f

File tree

-2

lines changed

-2

lines changed

Lines changed: 4 additions & 2 deletions

Original file line number	Diff line number	Diff line change
`@@ -126,8 +126,10 @@ print_var SRC_PATH_XLA`
`126`	`126`	`echo"=================================================="`
`127`	`127`
`128`	`128`	`# Parse SM_LIST into the format accepted by TransformerEngine's build system`
`129`		`-# "1.2,3.4,5.6" -> "12-real;34-real;56", i.e. SASS plus PTX for the last one`
`130`		`-NVTE_CUDA_ARCHS="${SM_LIST//,/-real;}"`
	`129`	`+# "1.2,3.4,5.6" -> "12;34;56". In principle we would like to compile SASS-only`
	`130`	`+# plus PTX for the highest known architecture, but TransformerEngine's build`
	`131`	`+# system does not currently handle that.`
	`132`	`+NVTE_CUDA_ARCHS="${SM_LIST//,/;}"`
`131`	`133`	`set -x`
`132`	`134`	`export NVTE_CUDA_ARCHS="${NVTE_CUDA_ARCHS//./}"`
`133`	`135`	`# Parallelism within nvcc invocations.`

Comments

(0)