search for: device_ordinal

Displaying 3 results from an estimated 3 matches for "device_ordinal".

2016 Mar 09
2
RFC: Proposing an LLVM subproject for parallelism runtime and support libraries
...// supported.) se::Platform *platform = se::MultiPlatformManager::PlatformWithName("cuda").ValueOrDie(); // Get a StreamExecutor for the chosen Platform. Multiple devices are // supported, we indicate here that we want to run on device 0. const int device_ordinal = 0; se::StreamExecutor *executor = platform->ExecutorForDevice(device_ordinal).ValueOrDie(); // Create a MultiKernelLoaderSpec, which knows where to find the code for our // kernel. In this case, the code is stored in memory as a PTX string. // // Note t...
2016 Mar 09
2
RFC: Proposing an LLVM subproject for parallelism runtime and support libraries
...se::Platform *platform = > se::MultiPlatformManager::PlatformWithName("cuda").ValueOrDie(); > > // Get a StreamExecutor for the chosen Platform. Multiple devices are > // supported, we indicate here that we want to run on device 0. > const int device_ordinal = 0; > se::StreamExecutor *executor = > platform->ExecutorForDevice(device_ordinal).ValueOrDie(); > > // Create a MultiKernelLoaderSpec, which knows where to find the > code for our > // kernel. In this case, the code is stored in memory as a PTX &gt...
2016 Mar 10
2
RFC: Proposing an LLVM subproject for parallelism runtime and support libraries
...gt;> se::MultiPlatformManager::PlatformWithName("cuda").ValueOrDie(); >> >> // Get a StreamExecutor for the chosen Platform. Multiple devices >> are >> // supported, we indicate here that we want to run on device 0. >> const int device_ordinal = 0; >> se::StreamExecutor *executor = >> platform->ExecutorForDevice(device_ordinal).ValueOrDie(); >> >> // Create a MultiKernelLoaderSpec, which knows where to find the >> code for our >> // kernel. In this case, the code is store...