frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 13:44:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

Trying something simpler.

Browse Source
This commit is contained in:
Clement Farabet
2012-09-05 00:19:47 -04:00
parent a8aef4e9ee
commit de87f835bb
1 changed files with 9 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
THCTensorRandom.cu
View File

@ -23,9 +23,15 @@ __host__ unsigned long THCRandom_seed()
__host__ void THCRandom_manualSeed(unsigned long the_seed_) __host__ void THCRandom_manualSeed(unsigned long the_seed_)
{ {
the_initial_seed = the_seed_; the_initial_seed = the_seed_;
if (rng==NULL) rng = new thrust::minstd_rand(the_initial_seed); if (initf == 0) {
else rng->seed(the_initial_seed); cudaMalloc(&rng, sizeof(thrust::minstd_rand));
initf = 1; thrust::minstd_rand rnghost(the_initial_seed);
cudaMemcpy(rng, &rnghost, sizeof(thrust::minstd_rand), cudaMemcpyHostToDevice);
cudaThreadSynchronize();
initf = 1;
} else {
rng->seed(the_initial_seed);
}
} }
__host__ unsigned long THCRandom_initialSeed() __host__ unsigned long THCRandom_initialSeed()