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

Commit 4cd991c

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

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
@@ -700,6 +700,10 @@ PYBIND11_MODULE(tclib, m) {
700700
&tc::CudaMappingOptions::unrollCopyShared,
701701
"Also unroll the copies to and from shared memory. If an unroll "
702702
"value is not provided, has no effect")
703+
.def(
704+
"privateDepth",
705+
&tc::CudaMappingOptions::privateDepth,
706+
"Specify the private depth")
703707
.def(
704708
"useReadOnlyCache",
705709
&tc::CudaMappingOptions::useReadOnlyCache,

0 commit comments

Comments
 (0)