MekkCyber
commited on
Commit
·
63184de
1
Parent(s):
70ebabf
torch extension rm
Browse files- layer-norm/ln_api.cpp +1 -1
layer-norm/ln_api.cpp
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
#include <torch/
|
| 2 |
#include "ATen/cuda/CUDAContext.h"
|
| 3 |
#include <c10/cuda/CUDAGuard.h>
|
| 4 |
|
|
|
|
| 1 |
+
#include <torch/torch.h>
|
| 2 |
#include "ATen/cuda/CUDAContext.h"
|
| 3 |
#include <c10/cuda/CUDAGuard.h>
|
| 4 |
|