Skip to content

Commit

Permalink
Fix style
Browse files Browse the repository at this point in the history
  • Loading branch information
nhatdongdang committed Jul 5, 2024
1 parent c1639bc commit e8db37a
Showing 1 changed file with 0 additions and 3 deletions.
3 changes: 0 additions & 3 deletions src/main.cu
Original file line number Diff line number Diff line change
Expand Up @@ -109,14 +109,12 @@ __device__ void propagate_fwd(matrix* weights, matrix* input_layer, matrix* outp
matrix_add(output_layer->data, biases->data, biases->rows);
}

// __global__ void infer(int a) {
__global__ void infer(matrix** d_inputs, int* d_results, matrix** d_weights, matrix** d_biases, int it_per_input, int in_num) {
int num_threads = blockDim.x * gridDim.x;
int thread_idx = (blockIdx.x * blockDim.x + threadIdx.x);

if (thread_idx > it_per_input) return;

// printf("Thread %d at input %d\n", thread_idx, in_num);

matrix* input = d_inputs[in_num];

Expand Down Expand Up @@ -178,7 +176,6 @@ int main(int argc, char* argv[]) {
weights[5] = new_matrix(40, 25);
weights[6] = new_matrix(52, 40);


biases[0] = new_matrix(98, 1);
biases[1] = new_matrix(65, 1);
biases[2] = new_matrix(50, 1);
Expand Down

0 comments on commit e8db37a

Please sign in to comment.