Skip to content

Commit c236adb

Browse files
authored
Merge pull request #126 from lcskrishna/cl/enable-setrng
enable set_rngstate function.
2 parents 90a9482 + fb162c6 commit c236adb

File tree

1 file changed

+1
-7
lines changed

1 file changed

+1
-7
lines changed

tools/amd_build/disabled_features.yaml

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -75,6 +75,7 @@
7575
"struct curandStateMtgp32*": "curandStateMtgp32*",
7676
"__host__ void THCRandom_getRNGState": "extern \"C\" __host__ void THCRandom_getRNGState",
7777
"__host__ void THCRandom_setRNGState": "extern \"C\" __host__ void THCRandom_setRNGState",
78+
"state[threadIdx.x].k = kernel;" : "state[threadIdx.x].set_params(kernel);"
7879
}
7980
},
8081
{
@@ -205,13 +206,6 @@
205206
"functions": [
206207
"THCTensor_(getTextureObject)"
207208
]
208-
},
209-
{
210-
"path": "aten/src/THC/THCTensorRandom.cu",
211-
"functions": [
212-
"THCRandom_setRNGState",
213-
"set_rngstate_kernel"
214-
]
215209
}
216210
]
217211
}

0 commit comments

Comments
 (0)