Definition at line 134 of file PrecomputeTask.h. References Impala::Core::Training::ComputeKernelMatrix(), and ILOG_INFO. Referenced by ComputeDevelKernel(), and ComputeTestKernel(). 00135 { 00136 //#ifdef KOENFIX 00137 //#define CPUFLOAT double 00138 //#else 00139 typedef float CPUFLOAT; 00140 //#endif 00141 00142 bool GPU = Link::Cuda::CudaUsed(); 00143 ILOG_INFO("GPU mode: " << GPU); 00144 int slabWidth = 1024; 00145 using Training::ComputeKernelMatrix; 00146 if (GPU) 00147 { 00148 ComputeKernelMatrix<float>(pt, slabWidth, GPU); 00149 } 00150 else 00151 { 00152 #ifdef CUDA 00153 ComputeKernelMatrix<CPUFLOAT>(pt, slabWidth, GPU); 00154 #else 00155 ComputeKernelMatrix<double>(pt, slabWidth, GPU); 00156 #endif 00157 } 00158 delete pt; 00159 }
Here is the call graph for this function:
|