Skip to content
This repository was archived by the owner on Apr 28, 2023. It is now read-only.

Commit 4807746

Browse files
author
Jules Pondard
committed
Add privateDepth function to PyBinds
Allow the user to set manually the new option privateDepth.
1 parent ab0982f commit 4807746

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tensor_comprehensions/pybinds/tclib.cc

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -680,6 +680,10 @@ PYBIND11_MODULE(tclib, m) {
680680
&tc::CudaMappingOptions::unrollCopyShared,
681681
"Also unroll the copies to and from shared memory. If an unroll "
682682
"value is not provided, has no effect")
683+
.def(
684+
"privateDepth",
685+
&tc::CudaMappingOptions::privateDepth,
686+
"Specify the private depth")
683687
.def(
684688
"useReadOnlyCache",
685689
&tc::CudaMappingOptions::useReadOnlyCache,

0 commit comments

Comments
 (0)