Skip to content

GitLab

Commit fa8e7d99 authored by Thor Johnsen's avatar Thor Johnsen
Browse files

Clean up code

parent 05dd9c69
Expand all Hide whitespace changes
Inline Side-by-side
Showing with 1884 additions and 1773 deletions
apex/contrib/csrc/bottleneck/bottleneck.cpp
View file @ fa8e7d99
This diff is collapsed.
apex/contrib/csrc/peer_memory/peer_memory_cuda.cu
View file @ fa8e7d99
...@@ -388,7 +388,7 @@ void push_pull_halos_1d( ...@@ -388,7 +388,7 @@ void push_pull_halos_1d(
const int numThreads = 128; const int numThreads = 128;
dim3 block(numThreads,1,1); dim3 block(numThreads,1,1);
AT_DISPATCH_ALL_TYPES_AND(at::ScalarType::Half, top_out_halo.scalar_type(), "push_pull_halos_1d_kernel", [&]{ AT_DISPATCH_ALL_TYPES_AND(at::ScalarType::Half, top_out_halo.scalar_type(), "push_pull_halos_1d_kernel", [&]{
if (diagnostics) printf("size(scalar_t) = %d\n",sizeof(scalar_t)); if (diagnostics) printf("size(scalar_t) = %ld\n",sizeof(scalar_t));
scalar_t* toh_p = top_out_halo.data_ptr<scalar_t>(); scalar_t* toh_p = top_out_halo.data_ptr<scalar_t>();
scalar_t* tox_p = top_out_tx.data_ptr<scalar_t>(); scalar_t* tox_p = top_out_tx.data_ptr<scalar_t>();
scalar_t* tix_p = top_inp_tx.data_ptr<scalar_t>(); scalar_t* tix_p = top_inp_tx.data_ptr<scalar_t>();
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment