Commit 67ea635f authored by aiss's avatar aiss
Browse files

push dsv0.8.2 version

parent 1b2721ad
Pipeline #201 failed with stages
in 0 seconds
/*
Copyright The Microsoft DeepSpeed Team
*/
#pragma once #pragma once
#include <cuda.h> #include <cuda.h>
......
/*
Copyright The Microsoft DeepSpeed Team
*/
#pragma once #pragma once
#include <cuda.h> #include <cuda.h>
......
/*
Copyright The Microsoft DeepSpeed Team
*/
/* Taken from NVIDIA/apex commit 855808f3fc268e9715d613f3c2e56469d8c986d8 */ /* Taken from NVIDIA/apex commit 855808f3fc268e9715d613f3c2e56469d8c986d8 */
#include <ATen/ATen.h> #include <ATen/ATen.h>
......
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
/*
Copyright The Microsoft DeepSpeed Team
*/
#include "cublas_wrappers.h" #include "cublas_wrappers.h"
#ifdef __HIP_PLATFORM_HCC__ #ifdef __HIP_PLATFORM_HCC__
......
/*
Copyright The Microsoft DeepSpeed Team
*/
#include "custom_cuda_layers.h" #include "custom_cuda_layers.h"
const int unroll_factor = 4; const int unroll_factor = 4;
......
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
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